Monk Object Detection's 2_pytorch_finetune
Project description
Project Details
Pipeline based on GluonCV Fintuning project - https://gluon-cv.mxnet.io/build/examples_detection/index.html
Installation
Supports
- Python 3.6
- Python 3.7
cd installation
Check the cuda version using the command
nvcc -V
Select the right requirements file and run
cat <selected requirements file> | xargs -n 1 -L 1 pip install
For example for cuda 9.0
cat requirements_cuda9.0.txt | xargs -n 1 -L 1 pip install
Functional Documentation
Pipeline
- Load Dataset
gtf.Dataset(root_dir, img_dir, anno_file, batch_size=batch_size);
- Load Model
gtf.Model(model_name, use_pretrained=pretrained, use_gpu=gpu);
- Set Hyper-parameter
gtf.Set_Learning_Rate(0.001);
- Train
gtf.Train(epochs, params_file);
TODO
- Add SSD support
- Add YoloV3 support
- Add support for Coco-Type Annotated Datasets
- Add support for VOC-Type Annotated Dataset
- Add Faster-RCNN support
- Test on Kaggle and Colab
- Add validation feature & data pipeline
- Add Optimizer selection feature
- Enable Learning-Rate Scheduler Support
- Enable Layer Freezing
- Set Verbosity Levels
- Add Project management and version control support (Similar to Monk Classification)
- Add Graph Visualization Support
- Enable batch proessing at inference
- Add feature for top-k output visualization
- Add Multi-GPU training
- Auto correct missing or corrupt images - Currently skips them
- Add Experimental Data Analysis Feature
External Contributors list
- https://github.com/THEFASHIONGEEK: Multi GPU feature
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
monk_obj_test2-0.0.2.tar.gz
(18.2 kB
view hashes)
Built Distribution
Close
Hashes for monk_obj_test2-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6c5bff8c14ba48b982e687ee727f921703f5c2293439e009ee3b761413df9289 |
|
MD5 | d361ce8017dfb36375402bc970785724 |
|
BLAKE2b-256 | acd0d456dbe5c3d6aeea8459e8bf1b35d59a079e76a5722ab9b7a84cd24c22c4 |