search

ShishirPatil / poet

ML model training for edge devices
https://shishirpatil.github.io/poet/
Apache License 2.0
157 stars 18 forks source link

May I know how to apply POET to YOLOv5 #1

Closed chaoyanghe closed 1 year ago

chaoyanghe commented 2 years ago

Dear Authors,

I like this work very much. May I know how to apply it on YOLOv5 (https://github.com/ultralytics/yolov5)?

ShishirPatil commented 2 years ago

Hi @chaoyanghe, thanks for checking out the project!

Apologies for the delay in responding. When we open source POET (very soon!) we will make sure to include many layers which can be used as building blocks for NN architectures. We will also include example architectures so that adding additional models (e.g., YOLOv5 in this case), should be relatively straightforward!

I'll keep this issue open, so you can circle back if you happen to run in to any model specific concerns once we OSS.

Thanks! Shishir

ShishirPatil commented 1 year ago

Hi @chaoyanghe,

We have open sourced POET. And you can look at poet/architectures for inspiration. I'm closing the issue but let us know if you have any further questions or need any help with your implementation.

Thanks!