All files are in the root folder.
config.py
contains configuration to edit the model for YoloV3dataset.py
contains the preprocessing of data into MindRecord, data augmentation operations, and preparing data for YoloV3 training.utils.py
contains utilies to calculate iou, suppression of overlapping bounding boxes with Non-Maximal Suppression(nms) and calculate precision and recall values.yolov3.py
contains the model structure.
Only training on ModelArts are currently supported, you are free to modify the code to enable GPU training, but the operation might not be supported yet.
The file train.py allows a person to train the model, it will download the files(on OBS) using Moxing Framework and processes them.
You need to copy all the files in this directory onto a OBS bucket, place them in a folder called src
.
The process of training a ModelArts is as below please follow the Model Training and Guidlines, the parameters for the creating the algorithm and training job is as follow:
- Parameters for creating an algorithm:
- Parameters for creating an