Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,239 Commits 1 Branch 0 Tags
5bab9a28e45f119839c14d91dc93bbdedadaf7de
Commit Graph
1239 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 5bab9a28e4 Reformat (#3694) 2021-06-19 14:21:28 +02:00
lb-desupervisedandLewis Belcher bfb2276b1d Slightly modify CLI execution (#3687) 
* Slightly modify CLI execution

This simple change makes it easier to run the primary functions of this
repo (train/detect/test) from within Python. An object which represents
`opt` can be constructed and fed to the `main` function of each of these
modules, rather than having to call the lower level functions directly,
or run the module as a script.

* Update export.py

Add CLI parsing update for more convenient module usage within Python.

Co-authored-by: Lewis Belcher <lb@desupervised.io>
 2021-06-19 12:06:59 +02:00
Mai Thanh MinhandGlenn Jocher bf209f6fe9 Skip HSV augmentation when hyperparameters are [0, 0, 0] (#3686) 
* Create shortcircuit in augment_hsv when hyperparameter are zero

* implement faster opt-in

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-19 11:51:21 +02:00
Glenn Jocher 814806c61d Update cache check (#3691) 
Swapped order of operations for faster first per https://github.com/ultralytics/yolov5/commit/f527704cd32c42bc0bba9cce04601783b8563204#r52362419
 2021-06-19 11:22:09 +02:00
Glenn Jocher 463628a4d8 Update README.md (#3684) 2021-06-18 17:12:42 +02:00
ZouJiu1andGlenn Jocher 2729761458 EMA changes for pre-model's batch_size (#3681) 
* EMA changes for pre-model's batch_size

* Update train.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-18 16:26:52 +02:00
Glenn Jocher f527704cd3 Cache v0.3: improved corrupt image/label reporting (#3676) 
* Cache v0.3: improved corrupt image/label reporting

Fix for https://github.com/ultralytics/yolov5/issues/3656#issuecomment-863660899

* cleanup
 2021-06-18 10:21:47 +02:00
Glenn Jocher 2296f1546f Update WORLD_SIZE and RANK retrieval (#3670) 2021-06-17 23:24:30 +02:00
Glenn Jocher 045d5d8629 Update TensorBoard (#3669) 2021-06-17 22:12:42 +02:00
Glenn Jocher fa201f968e Update train(hyp, *args) to accept hyp file or dict (#3668) 2021-06-17 22:03:25 +02:00
Glenn Jocher 6d6e2ca65f Update train.py (#3667) 2021-06-17 21:32:39 +02:00
Glenn Jocher ac34834563 Update download_weights.sh comment (#3662) 2021-06-17 16:21:10 +02:00
Glenn Jocher 2754adad46 Remove /weights directory (#3659) 
* Remove `/weights` directory

* cleanup
 2021-06-17 14:44:12 +02:00
Glenn Jocher 9b6dba6207 Update dataset_stats() to list of dicts (#3657) 
* Update `dataset_stats()` to list of dicts

@KalenMike

* Update datasets.py
 2021-06-17 13:59:52 +02:00
SpongeBab df7706d802 Update README.md (#3650) 
Be more user-friendly to new users
 2021-06-17 12:37:53 +02:00
Mai Thanh Minh fa29125f18 is_coco list fix (#3646) 2021-06-16 22:56:16 +02:00
Glenn Jocher 6187edcb53 Update CONTRIBUTING.md (#3647) 2021-06-16 19:57:11 +02:00
Glenn Jocher 3ce0db89b0 Update CONTRIBUTING.md (#3645) 
* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md
 2021-06-16 19:48:30 +02:00
xiaowk5516andGlenn Jocher d808855f77 Assert non-premature end of JPEG images (#3638) 
* premature end of JPEG images

* PEP8 reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-16 13:31:26 +02:00
Glenn Jocher bb79e13d52 RUN pip install --no-cache -U torch torchvision (#3637) 2021-06-16 12:35:33 +02:00
Glenn Jocher 6c0e1d9fd7 Update verify_image_label() (#3635) 2021-06-16 11:12:15 +02:00
Glenn Jocher 65f81bfefa Add **/*.torchscript.pt (#3634) 2021-06-16 10:49:50 +02:00
Glenn Jocher de56813ba8 FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3633) 2021-06-16 10:40:31 +02:00
Glenn Jocher 30e4c4f092 Update README.md (#3624) 2021-06-15 15:44:10 +02:00
Glenn Jocher 7d3686a686 Update check_file() (#3622) 
* Update `check_file()`

* Update datasets.py
 2021-06-15 13:21:04 +02:00
Wei Quan 4c5d9bff80 Fix incorrect end epoch comment (#3612) 2021-06-15 11:24:56 +02:00
Ayush ChaurasiaandGlenn Jocher daab682b06 Suppress wandb images size mismatch warning (#3611) 
* supress wandb images size mismatch warning

* supress wandb images size mismatch warning

* PEP8 reformat and optimize imports

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-14 18:54:58 +02:00
Glenn Jocher 239a11c197 Download COCO and VOC by default (#3608) 2021-06-14 13:58:54 +02:00
masood azharandGlenn Jocher 6062319ec3 Update datasets.py (#3591) 
* 'changes-in_dataset'

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-14 12:28:18 +02:00
Glenn Jocher f8adee156d merge develop 2021-06-14 10:45:15 +02:00
Kalen MichaelandGlenn Jocher 3133607560 Simplify README.md (#3530) 
* Update README.md

* added hosted images

* added new logo

* testing image hosting

* changed svgs to pngs

* removed old header

* Update README.md

* correct colab image source

* splash.jpg

* rocket and W&B fix

* added contributing template

* added social media to top section

* increased size of top social media

* cleanup and updates

* rearrange quickstarts

* API cleanup

* PyTorch Hub cleanup

* Add tutorials

* cleanup

* update CONTRIBUTING.md

* Update README.md

* update wandb link

* Update README.md

* remove tutorials header

* update environments and integrations

* Comment API image

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* double spaces after section

* Update README.md

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-13 02:37:20 +02:00
Glenn Jocher 88b1945241 Delete __init__.py (#3596) 2021-06-12 15:21:37 +02:00
Glenn Jocher 7a565f130a Update dataset_stats() (#3593) 
@KalenMike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.

Usage example:
```python
from utils.datasets import *

dataset_stats('coco128.yaml', verbose=True)
```
 2021-06-12 13:26:41 +02:00
Ayush Chaurasia 53d4fc2e26 W&B: Allow changed in config variable #3588 2021-06-11 23:18:57 +02:00
Glenn Jocher 4984cf54be train.py GPU memory fix (#3590) 
* train.py GPU memory fix

* ema

* cuda

* cuda

* zeros input

* to device

* batch index 0
 2021-06-11 20:24:03 +02:00
Glenn Jocher e8c5237403 ConfusionMatrix normalize=True fix (#3587) 2021-06-11 11:46:05 +02:00
Glenn Jocher ec2da4a82c Add ConfusionMatrix normalize=True flag (#3586) 2021-06-11 11:37:08 +02:00
Glenn Jocher 46e1fdfbc6 Update stale.yml (#3585) 2021-06-11 11:23:39 +02:00
Glenn Jocher 5c32bd3080 Created using Colaboratory 2021-06-10 16:01:34 +02:00
Glenn Jocher 53ed872c28 Update export.py, yolo.py sys.path.append() (#3579) 2021-06-10 15:35:22 +02:00
Glenn Jocher 095197bd4a Ignore Seaborn plot warnings (#3576) 
* Ignore Seaborn plot warnings

* Update plots.py

* Update metrics.py
 2021-06-10 13:51:29 +02:00
Glenn Jocher 4695ca8314 Refactoring cleanup (#3565) 
* Refactoring cleanup

* Update test.py
 2021-06-09 22:50:27 +02:00
Glenn Jocher 0e5cfdbea7 Refactor models/export.py arguments (#3564) 
* Refactor models/export.py arguments

* cleanup

* cleanup
 2021-06-09 22:43:46 +02:00
Glenn Jocher 66cf5c28c1 Refactor detect.py arguments (#3559) 
* Refactor detect.py arguments

@SkalskiP @KalenMike

* unused ok

* comment arguments
 2021-06-09 22:19:34 +02:00
Glenn Jocher a9553c04a7 Refactor test.py arguments (#3558) 
* remove opt from test()

* pass kwargs

* update comments

* revert accidental default change

* multiple --img options

* add comments
 2021-06-09 21:36:10 +02:00
dependabot[bot] c6deb73a89 Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 
Bumps [pip](https://github.com/pypa/pip) from 18.1 to 19.2.
- [Release notes](https://github.com/pypa/pip/releases)
- [Changelog](https://github.com/pypa/pip/blob/main/NEWS.rst)
- [Commits](https://github.com/pypa/pip/compare/18.1...19.2)

---
updated-dependencies:
- dependency-name: pip
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-06-09 21:14:09 +02:00
Glenn Jocher 4ef9261870 Remove redundant speed/study half argument (#3557) 2021-06-09 16:30:01 +02:00
Glenn Jocher 5948f20a3d Update test.py profiling (#3555) 
* Update test.py profiling

* half_precision to half

* inplace
 2021-06-09 16:25:17 +02:00
Glenn Jocher 8b5086c21b Multi-GPU default to single device 0 (#3554) 
* Multi-GPU default to single device 0

* Multi-GPU default to single device 0

* add space
 2021-06-09 15:31:47 +02:00
Glenn Jocher 63157d214d Remove is_coco argument from test() (#3553) 2021-06-09 15:09:51 +02:00