Skip to content

Darkshadow9799/shadowCut

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 

Repository files navigation

shadowCut

  • 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 as ResNet152V2(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)

To Get Started:

  1. Import the library: import shadowCut as sc
  2. Get the training folder address and testing folder address. For example: train = './data/train' and
    test = './data/test'
  3. 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)
  4. 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)

For Contribution:

  1. General script for all type of Generative Adverserial Network which can be excessed for by any new starter (and also experienced).
  2. 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).
  3. 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.

About

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages