search

ultralytics / yolov3

YOLOv3 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
10.18k stars 3.44k forks source link

Add to PyTorch model hub with hubconf.py #1138

Closed pcgeek86 closed 4 years ago

pcgeek86 commented 4 years ago

🚀 Feature

Motivation

In order to make this model easier to find and interact with, consider publishing it to the PyTorch model hub.

https://pytorch.org/hub/

Additional context

This articles describes how to publish a model to the PyTorch model hub.

https://pytorch.org/blog/towards-reproducible-research-with-pytorch-hub/

Once hubconf.py is created in the yolov3 repository, a pull request must be sent to the PyTorch Hub repository, which registers this repo with the Hub: https://github.com/pytorch/hub/blob/master/docs/template.md

github-actions[bot] commented 4 years ago

Hello @pcgeek86, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Google Colab Notebook, Docker Image, and GCP Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

glenn-jocher commented 4 years ago

@pcgeek86 that's an excellent suggestion, thanks for the feature request! Would you want to take the lead on this by submitting a PR with the required changes?

pcgeek86 commented 4 years ago

Hi @glenn-jocher, yes I would be happy to take this on. I was also planning to send in some other small pull requests, as I've encountered some issues on Windows 10. I was able to work around everything just fine, thankfully!

Keep in mind that I'm not very experienced with Python, so please be patient with my contribution. It'll be great to see YOLOv3 in the PyTorch model hub!

glenn-jocher commented 4 years ago

@pcgeek86 great! I've tried to make the code as OS-agnostic as possible, but if you find any windows bugs that you can smooth over yes please let me know or submit a PR. Thanks!

github-actions[bot] commented 4 years ago

This issue is stale because it has been open 30 days with no activity. Remove Stale label or comment or this will be closed in 5 days.