- Load foldered dataset
- Set number of epochs
- Run training
ptf = prototype(verbose=1)
ptf.Prototype("sample-project-1", "sample-experiment-1")
ptf.Default(dataset_path="./dataset_cats_dogs_train/",
model_name="resnet18", freeze_base_network=True, num_epochs=2)
ptf.Train()
img_name = "./monk/datasets/test/0.jpg";
predictions = ptf.Infer(img_name=img_name, return_raw=True);
print(predictions)
- Add created experiments with different hyperparameters
- Generate comparison plots
ctf = compare(verbose=1);
ctf.Comparison("Sample-Comparison-1");
ctf.Add_Experiment("sample-project-1", "sample-experiment-1");
ctf.Add_Experiment("sample-project-1", "sample-experiment-2");
.
.
.
ctf.Generate_Statistics();
Support for
- OS
- Ubuntu 16.04
- Ubuntu 18.04
- Mac OS
- Windows
- Python
- Version 3.6
- Version 3.7
- Cuda
- Version 9.0
- Version 9.2
- Version 10.0
- Version 10.1
For Installation instructions visit: Link
-
- Functional Documentation (Will be merged with Latest docs soon)
-
- Features and Functions (In development):
-
- Complete Latest Docs (In Progress)
- Model Visualization
- Pre-processed data visualization
- Learned feature visualization
- NDimensional data input - npy - hdf5 - dicom - tiff
- Multi-label Image Classification
- Custom model development
- Incorporate pep coding standards
- Functional Documentation
- Tackle Multiple versions of libraries
- Add unit-testing
- Contribution guidelines
- Tensorflow 2.0
- Chainer
- TensorRT Acceleration
- Intel Acceleration
- Echo AI - for Activation functions
Copyright 2019 onwards, Tessellate Imaging Private Limited Licensed under the Apache License, Version 2.0 (the "License"); you may not use this project's files except in compliance with the License. A copy of the License is provided in the LICENSE file in this repository.