search

keras-team / keras-cv

Industry-strength Computer Vision workflows with Keras
Other
1.01k stars 330 forks source link

Implement auto anchor algorithm #1257

Open LukeWood opened 1 year ago

LukeWood commented 1 year ago

Reference: https://github.com/ultralytics/yolov5/blob/master/utils/autoanchor.py

innat commented 1 year ago

https://github.com/google/automl/issues/412

bhack commented 1 year ago

There was a PR at https://github.com/tensorflow/models/pull/10367

bhack commented 1 year ago

https://github.com/tensorflow/models/blob/master/official/projects/yolo/ops/kmeans_anchors.py

https://github.com/tensorflow/models/blob/master/official/projects/yolo/ops/kmeans_anchors_test.py

LukeWood commented 1 year ago

Thanks for the links @bhack !

LukeWood commented 1 year ago

(Just so I don't lose my colab, here is my temporary working copy: https://colab.research.google.com/drive/1WjwV45Nz9EWEFV7kP-aMLmE2-h0RnHZl?usp=sharing)

not mean't to be public yet.