Skip to content
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.

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

Advice about parameters #572

Closed
Vlad15lav opened this issue Dec 24, 2020 · 2 comments
Closed

Advice about parameters #572

Vlad15lav opened this issue Dec 24, 2020 · 2 comments

Comments

@Vlad15lav
Copy link

@zylo117 Hi, thanks for the repository. I use the same dataset #569, but all 20 classes. I have already trained this dataset on Faster R-CNN, SSD, YOLOv3 and there was a good result (65-75) mAP, but on this model 43 mAP.
link: https://arxiv.org/ftp/arxiv/papers/1909/1909.00133.pdf

I used the K-means with parameters anchor_base_scale = 4, anchor_stride = 8.

Accuracy: 48.42%
Boxes:
 [[ 36.48  37.44]
 [ 13.44  13.44]
 [142.08 159.36]]
computed paras:  ([1.1399999999999997, 0.41999999999999993, 4.440000000000001], [(1, 1.026315789473686), (1, 1.0), (1, 1.1216216216216213)])

I watched the tutorial #569 and tried learning in two stages.
python train.py -c 2 -p dior --head_only True --lr 5e-4 --batch_size 4 --load_weights weights/efficientdet-d2.pth --num_epochs 100 --save_interval 1100
image

python train.py -c 2 -p dior --head_only False --lr 1e-4 --batch_size 4 --load_weights last --num_epochs 100 --save_interval 1100
image

What can I do to improve the mAP?

@Vlad15lav Vlad15lav changed the title How to boost mAP Advice about parameters Dec 29, 2020
@Vlad15lav
Copy link
Author

@zylo117 I have 9 anchors (accuracy 69) for Yolov3, can I install them on this repository? If I use K means for 3 clusters then I get a bad result (accuracy 47).

@zylo117
Copy link
Owner

zylo117 commented Dec 29, 2020

Try creating your own anchors with this. It works for me.
#308

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants