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

Pycocotools best.pt after COCO train #1616

Merged
merged 2 commits into from
Dec 6, 2020
Merged

Pycocotools best.pt after COCO train #1616

merged 2 commits into from
Dec 6, 2020

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Dec 6, 2020

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Implementation of YOLOv3 and YOLOv3-tiny configurations into the Ultralytics yolov5 repository and refinement of various utilities.

πŸ“Š Key Changes

  • Added YOLOv3-tiny and YOLOv3 model configurations to the project.
  • Removed unnecessary usage of glob in test.py, specifying the exact annotation JSON path instead.
  • Enhanced string formatting in Python by utilizing f-strings in test.py.
  • Updated train.py with new padding strategy and streamlined optimizer state saving.
  • Made a small change in google_utils.py to ensure weight strings are properly formatted.

🎯 Purpose & Impact

  • πŸ— The new model configurations allow researchers and developers to train YOLOv3 and YOLOv3-tiny models with the Ultralytics framework, offering more variety and flexibility.
  • πŸ›  The update enhances code clarity and maintainability by specifying exact file paths and adopting modern Python string formatting strategies.
  • πŸš€ The train.py adjustments aim to optimize the training process, potentially leading to better performance or faster training times.
  • πŸ‘©β€πŸ’» The changes in google_utils.py aim to minimize errors related to weight file handling, leading to a smoother user experience when downloading pre-trained models.

@glenn-jocher
Copy link
Member Author

This PR runs best.pt with pycocotools after COCO training. Custom training is unaffected. Additional updates were made to testloader in train.py to match exactly the arguments used in test.py when called directly. This should lead to more reproducible results between train.py and test.py, which a number of users had raised issues about previously.

@glenn-jocher
Copy link
Member Author

Lastly YOLOv3 models are imported to models/hub from the newly updated https://github.com/ultralytics/yolov3.

@glenn-jocher glenn-jocher merged commit 791dadb into master Dec 6, 2020
@glenn-jocher glenn-jocher deleted the train_coco branch December 6, 2020 13:58
burglarhobbit pushed a commit to burglarhobbit/yolov5 that referenced this pull request Jan 1, 2021
* Pycocotools best.pt after COCO train

* cleanup
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
* Pycocotools best.pt after COCO train

* cleanup
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Pycocotools best.pt after COCO train

* cleanup
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

Successfully merging this pull request may close these issues.

1 participant