Ultralytics master #5
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
* Create GlobalWheat2020.yaml * Update and rename visdrone.yaml to VisDrone.yaml * Update GlobalWheat2020.yaml
* add object365 * ADD CONVERSION SCRIPT * fix transcript * Reformat and simplify * spelling * Update get_objects365.py Co-authored-by: Glenn Jocher <[email protected]>
* Added flag to enable/disable all inplace and assignment operations * Removed shape print statements * Scope Detect/Model import to avoid circular dependency * PEP8 * create _descale_pred() * replace lost space * replace list with tuple Co-authored-by: Glenn Jocher <[email protected]>
* Improve performance of Dataset Logger * Fix scaling bug
* Curl update * Curl update
* Added support for fp16 (half) to export.py * minimize code additions Co-authored-by: Glenn Jocher <[email protected]>
* Add NMS threshold checks * fix
…ralytics#2982) * Update yolo.py * Update export.py * fix export grid * Update export.py, remove detect export attribute * rearrange if order * remove --grid, default inplace=False * rename exp_dynamic to onnx_dynamic, comment * replace bs with 1 in anchor_grid[i] index 0 * Update export.py Co-authored-by: Glenn Jocher <[email protected]>
* download() ThreadPool update * update train image count * cid + 1
Issues with 21.04
) * Add get_coco128.sh * Update get_coco128.sh Co-authored-by: Glenn Jocher <[email protected]>
Co-authored-by: kepler62f <>
* Explicitly convert artifact path to posix_path * Remove redudant str() casting
* Allow users to skip exporting in formats that they don't care about * Correct comments * Update export.py renamed --skip-format to --exclude * Switched format from exclude to include (as instructed by @glenn-jocher) * cleanup Co-authored-by: Glenn Jocher <[email protected]>
* Scope all hubconf.py imports for torch.hub.list() * Update hubconf.py
* SKU-110K CVPR2019 Dataset Auto-Download This PR adds the **SKU-110K CVPR2019 Dataset** https://github.com/eg4000/SKU110K_CVPR19 to the list of YOLOv5 🚀 auto-download list. - [x] [COCO](https://cocodataset.org/#home) - [x] [COCO128](https://www.kaggle.com/ultralytics/coco128) - [x] [VOC](http://host.robots.ox.ac.uk/pascal/VOC/) - [x] [Argoverse-HD](http://www.cs.cmu.edu/~mengtial/proj/streaming/) - [x] [VisDrone2019-DET](https://github.com/VisDrone/VisDrone-Dataset) - [x] [GlobalWheat-2020](http://www.global-wheat.com/) - [ ] [KITTI](https://www.kaggle.com/twaldo/kitti-object-detection) - [x] [SKU110](https://github.com/eg4000/SKU110K_CVPR19) (grocery store items) - [ ] [CityScapes](https://www.cityscapes-dataset.com/) - [x] [Objects365](https://www.objects365.org/overview.html) - [ ] [OpenImages](https://storage.googleapis.com/openimages/web/index.html) ```yaml # SKU-110K retail items dataset https://github.com/eg4000/SKU110K_CVPR19 # Train command: python train.py --data SKU-110K.yaml # Default dataset location is next to YOLOv5: # /parent_folder # /datasets/SKU-110K # /yolov5 ``` * Update SKU-110K.yaml * Update SKU-110K.yaml
* rename class autoShape -> AutoShape follow other class naming convention * rename class autoShape -> AutoShape follow other classes' naming convention * rename class autoShape -> AutoShape
* Allow to manual selection of device for torchhub models * single line device nested torch.device(torch.device(device)) ok Co-authored-by: Glenn Jocher <[email protected]>
* Added max_det parameters in various places * 120 character line * PEP8 * 120 character line * Update inference default to 1000 instances * Update inference default to 1000 instances Co-authored-by: Glenn Jocher <[email protected]>
* Color can be none by default * `plot_one_box()` default `color=(128, 128, 128)` Co-authored-by: Glenn Jocher <[email protected]>
Cython required by pycocotools in certain cases, i.e. cocodataset/cocoapi#172
* Check CoreML models.train() mode * Update export.py
Bug fix to check batch_size divisibility of utilized CUDA device count vs total system CUDA device count.
* YouTube stream ending fix Properly terminates YouTube streams on video end. Should resolve issues ultralytics#2769 and ultralytics#3220. * Update datasets.py
…ltralytics#3264) * adds latest tag to match wandb defaults * adds entity handling, 'last' tag * fixes bug causing finished runs to resume * removes redundant "last" tag for wandb artifact
* Changing save_one_box Made to work with other changes to common.py * PEP8 and single line BGR Co-authored-by: Glenn Jocher <[email protected]>
Ultralytics REST API fix.
Add try: except clause on missing requirements install to catch install failures. Prompted by Ultralytics YOLOv5 API failure on Cython.
…ltralytics-master # Conflicts: # models/export.py # requirements.txt # train.py
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.