New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Cannot reproduce mAP with potholes data #20

Open

tylertroy opened this issue Mar 16, 2022 · 0 comments

tylertroy commented Mar 16, 2022 •

edited

Loading

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment