Open tylertroy opened 2 years ago
I ran the command as specified in the docs to train on an RTX3090 using the potholes dataset as follows.
python train.py --use-pretrain True --model-type p5 --dataset-type voc --dataset dataset/pothole_voc --num-classes 1 --class-names pothole.names --voc-train-set dataset_1,train --voc-val-set dataset_1,val --epochs 200 --batch-size 4 --multi-scale 320,352,384,416,448,480,512 --augment ssd_random_crop
However I cannot reproduce the mAP as presented.
Similarly the training loss looks quite different.
Any thoughts on what I might be doing wrong? Or perhaps there has been a material change to the repo since those results were posted.
Feedback appreciated and thanks for providing this good work to the community.
I ran the command as specified in the docs to train on an RTX3090 using the potholes dataset as follows.
However I cannot reproduce the mAP as presented.
Similarly the training loss looks quite different.
Any thoughts on what I might be doing wrong? Or perhaps there has been a material change to the repo since those results were posted.
Feedback appreciated and thanks for providing this good work to the community.