Skip to main content

A small example package

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

Link




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

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

monk_test1-0.0.7.tar.gz (12.4 kB view hashes)

Uploaded Source

Built Distribution

monk_test1-0.0.7-py3-none-any.whl (13.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page