-
A python based package for fastening the deep learning pipeline. User has to select the path for train data, validation data and model as mandatory but rest of the parameters are optional and for customized as needed.
-
All the tensorflow functionalities are Inherited in this library and major part of Library is constructed using tensorflow.
-
Pypi: shadowCut - Pypi
-
Installation command:
pip install shadowCut -
Now lets see at example for using pretrained model:
-
Here suppose we have train_path as
'./train', valid_path as'./test', and the model asResNet152V2(input_shape=[224,224]+[3],weights='imagenet',include_top=False) -
We can use this package by using:
shadowCut.preTrainedModel(train_path=train_path, valid_path=test_path, model=model) -
This will return model's history and model respectively.
-
Libraries used:
PIL, os, sys, tensorflow, glob2, (tensorflow-gpu)
- Import the library:
import shadowCut as sc - Get the training folder address and testing folder address. For example:
train = './data/train'and
test = './data/test' - If want to get pretrained model: For example want to train ResNet152V2 model
model = tf.keras.applications.ResNet152V2()- If want custom output model then can be passed else will take already present output model.
- Config custom options if required.
- Pass it into:
sc.preTrainedModel(train = train, test = test, model = model, * other custom parameters)
- If want to get custom model: For example want to train ResNet152V2 model
model = #Custom model- If want custom output model then can be passed else will take already present output model.
- This library will use gpu if the system has it.
- Config custom options if required.
- Pass it into:
sc.customTrainedModel(train = train, test = test, model = model, * other custom parameters)
- General script for all type of Generative Adverserial Network which can be excessed for by any new starter (and also experienced).
- List of GANs to be added with a general script:
- DCGAN
- Condiational GAN
- Stack GAN
- Info GAN
- Disco GAN
- Style GAN
- Cycle GAN (more can be added).
- Add Minmax and Wasserstein loss function.
Futher for merging your code into the stable branch, you can contact the respective creator Aayush Jain to validate and merge your pull request.