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

[Feature] Calculate anchor boxes by kmeans and GA #69

Open
1 task done
rehohoho opened this issue Jul 13, 2021 · 1 comment
Open
1 task done

[Feature] Calculate anchor boxes by kmeans and GA #69

rehohoho opened this issue Jul 13, 2021 · 1 comment
Labels
enhancement New feature or request

Comments

@rehohoho
Copy link
Collaborator

Prerequisites

Please answer the following question for yourself before submitting an issue.

  • I checked to make sure that this feature has not been requested already.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/official/vision/beta

2. Describe the feature you request

A clear and concise description of what you want to happen.
Calculate the anchor boxes to use for one-shot detector strategies according to labels, using

  • kmeans
  • genetic algorithm

Some understanding on the boxes:
https://towardsdatascience.com/anchor-boxes-the-key-to-quality-object-detection-ddf9d612d4f9
pjreddie/darknet#568

Some implementations on generating boxes using clustering studies:
official darknet: https://github.com/AlexeyAB/darknet/blob/master/scripts/gen_anchors.py
ultralytics: https://github.com/ultralytics/yolov5/blob/1093b1d765fd2d1d603affdbb1473554b396bf1c/utils/autoanchor.py
ultralytics: ultralytics/yolov5#3546
keras: https://github.com/experiencor/keras-yolo3/blob/master/gen_anchors.py

3. Additional context

Add any other context about the feature request here.

4. Are you willing to contribute it? (Yes or No)

Yes

@rehohoho rehohoho added the enhancement New feature or request label Jul 13, 2021
@rehohoho rehohoho self-assigned this Jul 13, 2021
@rehohoho
Copy link
Collaborator Author

rehohoho commented Jul 26, 2021

Alternatively use YOLOX, that supercedes YOLOv4 and YOLOv5 which does not use anchors.
https://github.com/Megvii-BaseDetection/YOLOX
https://arxiv.org/pdf/2107.08430v1.pdf

Improvements:

  • Anchor-free
  • Advanced label assignment strategy
  • End-to-end (no NMS)

@rehohoho rehohoho removed their assignment Aug 20, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant