Note
This Python sample is available only with our Professional plan.
Training Pipeline of Object Detection
This Python script allows you to train a Pytorch Object Detection model with preprocessed HDF5 data and labeled bounding boxes.
Expected Output
Object Detection training result:
checkpoints (models at different training stages)
log files
videos on test dataset
A test demo is shown below:
Setup & requirements
To run the script, you need:
path to the output folder
path to the training dataset:
a folder containing 3 sub folders, named
train
,val
,test
.each subfolder should contain one or multiple
h5
files and their corresponding<h5 filename>_bbox.npy
labels.a dictionary file named
label_map_dictionary.json
, which contains all the detection categories.
Note
To have a quick test of the pipeline, you can simply set the path of training dataset to “toy_problem”. This will train the model on Moving MNIST Dataset.
How to start
To run the script:
Linux
python3 train_detection.py /path/to/logging /path/to/dataset
Windows
python train_detection.py /path/to/logging /path/to/dataset
To find the full list of options, run:
Linux
python3 train_detection.py -h
Windows
python train_detection.py -h