Skip to content

aaronkwc/msc21_training_example

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MindSpore YoloV3 Model for MindSpore Challenge 21

All files are in the root folder.

  1. config.py contains configuration to edit the model for YoloV3
  2. dataset.py contains the preprocessing of data into MindRecord, data augmentation operations, and preparing data for YoloV3 training.
  3. utils.py contains utilies to calculate iou, suppression of overlapping bounding boxes with Non-Maximal Suppression(nms) and calculate precision and recall values.
  4. yolov3.py contains the model structure.

Training

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:

  1. Parameters for creating an algorithm:
  2. Parameters for creating an

Testing

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%