Glenn Jocher
fa201f968e
Update `train(hyp, *args)` to accept `hyp` file or dict ( #3668 )
3 years ago
Glenn Jocher
6d6e2ca65f
Update train.py ( #3667 )
3 years ago
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
3 years ago
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
3 years ago
Glenn Jocher
9b6dba6207
Update `dataset_stats()` to list of dicts ( #3657 )
* Update `dataset_stats()` to list of dicts
@KalenMike
* Update datasets.py
3 years ago
SpongeBab
df7706d802
Update README.md ( #3650 )
Be more user-friendly to new users
3 years ago
Mai Thanh Minh
fa29125f18
`is_coco` list fix ( #3646 )
3 years ago
Glenn Jocher
6187edcb53
Update CONTRIBUTING.md ( #3647 )
3 years ago
Glenn Jocher
3ce0db89b0
Update CONTRIBUTING.md ( #3645 )
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
3 years ago
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>
3 years ago
Glenn Jocher
bb79e13d52
RUN pip install --no-cache -U torch torchvision ( #3637 )
3 years ago
Glenn Jocher
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
3 years ago
Glenn Jocher
65f81bfefa
Add `**/*.torchscript.pt` ( #3634 )
3 years ago
Glenn Jocher
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 ( #3633 )
3 years ago
Glenn Jocher
30e4c4f092
Update README.md ( #3624 )
3 years ago
Glenn Jocher
7d3686a686
Update `check_file()` ( #3622 )
* Update `check_file()`
* Update datasets.py
3 years ago
Wei Quan
4c5d9bff80
Fix incorrect end epoch comment ( #3612 )
3 years ago
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>
3 years ago
Glenn Jocher
239a11c197
Download COCO and VOC by default ( #3608 )
3 years ago
masood azhar
6062319ec3
Update datasets.py ( #3591 )
* 'changes-in_dataset'
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
f8adee156d
merge develop
3 years ago
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>
3 years ago
Glenn Jocher
88b1945241
Delete __init__.py ( #3596 )
3 years ago
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)
```
3 years ago
Ayush Chaurasia
53d4fc2e26
W&B: Allow changed in config variable #3588
3 years ago
Glenn Jocher
4984cf54be
train.py GPU memory fix ( #3590 )
* train.py GPU memory fix
* ema
* cuda
* cuda
* zeros input
* to device
* batch index 0
3 years ago
Glenn Jocher
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
3 years ago
Glenn Jocher
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
3 years ago
Glenn Jocher
46e1fdfbc6
Update stale.yml ( #3585 )
3 years ago
Glenn Jocher
5c32bd3080
Created using Colaboratory
3 years ago
Glenn Jocher
53ed872c28
Update export.py, yolo.py `sys.path.append()` ( #3579 )
3 years ago
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 years ago
Glenn Jocher
4695ca8314
Refactoring cleanup ( #3565 )
* Refactoring cleanup
* Update test.py
3 years ago
Glenn Jocher
0e5cfdbea7
Refactor models/export.py arguments ( #3564 )
* Refactor models/export.py arguments
* cleanup
* cleanup
3 years ago
Glenn Jocher
66cf5c28c1
Refactor detect.py arguments ( #3559 )
* Refactor detect.py arguments
@SkalskiP @KalenMike
* unused ok
* comment arguments
3 years ago
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
3 years ago
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>
3 years ago
Glenn Jocher
4ef9261870
Remove redundant speed/study `half` argument ( #3557 )
3 years ago
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 years ago
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
3 years ago
Glenn Jocher
63157d214d
Remove `is_coco` argument from `test()` ( #3553 )
3 years ago
Glenn Jocher
958ab92dc1
Remove `opt` from `create_dataloader()`` ( #3552 )
3 years ago
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
3 years ago
Glenn Jocher
f8ec71e1c2
Update `check_python(minimum=3.6.2)` ( #3548 )
3 years ago
Glenn Jocher
ef0b5c9d29
On-demand `pycocotools` pip install ( #3547 )
3 years ago
Glenn Jocher
1b5edb6f8e
Update `dataset_stats()` for HUB ( #3536 )
* Update `dataset_stats()` for HUB
Cleanup of b6fdd2e
* autodownload flag
* Update general.py
* cleanup
3 years ago
Glenn Jocher
b6fdd2e5e5
Create `dataset_stats()` for HUB
3 years ago
Glenn Jocher
ac8691e208
Created using Colaboratory
3 years ago
Glenn Jocher
78cf488556
Created using Colaboratory
3 years ago
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
3 years ago