mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,229 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

1229 Commits (6d6e2ca65fbcc5c8de7f6a796ac462bed7a732c7)
 

Author SHA1 Message Date
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
xiaowk5516 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 Chaurasia 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 azhar 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 Michael 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
Glenn Jocher 958ab92dc1
Remove `opt` from `create_dataloader()`` (#3552) 2021-06-09 13:14:56 +02:00
Glenn Jocher 0cfc5b2c18
Update README.md (#3550) 
Add permanent splash URL and update hyperlink from iOS landing page to Ultralytics YOLOv5 landing page at https://ultralytics.com/yolov5
 2021-06-09 11:49:25 +02:00
Glenn Jocher f8ec71e1c2
Update `check_python(minimum=3.6.2)` (#3548) 2021-06-09 11:34:15 +02:00
Glenn Jocher ef0b5c9d29
On-demand `pycocotools` pip install (#3547) 2021-06-09 11:22:21 +02:00
Glenn Jocher 1b5edb6f8e
Update `dataset_stats()` for HUB (#3536) 
* Update `dataset_stats()` for HUB 

Cleanup of b6fdd2e

* autodownload flag

* Update general.py

* cleanup
 2021-06-09 10:56:11 +02:00
Glenn Jocher b6fdd2e5e5
Create `dataset_stats()` for HUB 2021-06-08 23:09:45 +02:00
Glenn Jocher ac8691e208 Created using Colaboratory 2021-06-08 19:01:52 +02:00
Glenn Jocher 78cf488556 Created using Colaboratory 2021-06-08 18:54:33 +02:00
Glenn Jocher c6b51f4189
Update FP16 `--half` argument for test.py and detect.py (#3532) 
* Update FP16 `--half` argument for test.py and detect.py

* Update detect.py
 2021-06-08 18:47:13 +02:00
Glenn Jocher 8d52c1c5c5
Update datasets.py (#3531) 
Minor updates to https://github.com/ultralytics/yolov5/pull/3505, inplace accumulation.
 2021-06-08 18:36:40 +02:00