andy-yun / pytorch-0.4-yolov3

Yet Another Implimentation of Pytroch 0.4.1 and YoloV3 on python3
MIT License
279 stars 72 forks source link

Getting training for (390407, 6101) #32

Closed Haosam closed 5 years ago

Haosam commented 5 years ago

I am trying to train yolo on my own dataset of 90 images of 2 classes. I have the .data file with the train.txt, test.txt, names all set to the absolute paths of the text files, and .name file. My train and test .txt files contain the absolute path to the images My .names file contains 2 classes and I am using the original yolov3.weights file I also have configured the .cfg to fit 2 classes

However when I train, with the code python train.py -d {.data file} -c {.cfg file} -w {yolov3.weights}

Only one line is displayed Training for (390407, 6101)

And the build ends. I do not see any training processes being done. Is there something wrong with what I did?

Haosam commented 5 years ago

python train.py -d {.data file} -c {.cfg file} -w {yolov3.weights} -r found the reason. Had to add a -r behind as I was training from scratch