CU-Robosub / Yolo-Train-System

Convolutional Neural Networks
http://pjreddie.com/darknet/
0 stars 0 forks source link

Script to Create Training Set From Zips #12

Closed skhadem closed 5 years ago

skhadem commented 5 years ago

We want to be able to run one script to turn zipped data sets into one large training set. What this needs to do:

  1. Put all images into one folder
  2. Put all labels into one folder
  3. Create train.txt with names of all images to train on
  4. Create test.txt with names of all images to validate with
  5. Configure obj.names with class names
  6. Configure obj.data with the paths to the data and number of classes
  7. (ideal, harder) configure yolo-obj.cfg to setup layer sizes, batch sizes, class number, etc.

See how to train