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

191 Commits (9dd0ad30c0668cc467b4496f45de2a85a901830d)

Author SHA1 Message Date
Glenn Jocher d3ea0df8b9
New YOLOv5 Classification Models (#8956) 
* Update

* Logger step fix: Increment step with epochs (#8654)

* enhance

* revert

* allow training from scratch

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update --img argument from train.py 

single line

* fix image size from 640 to 128

* suport custom dataloader and augmentation

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* format

* Update dataloaders.py

* Single line return, single line comment, remove unused argument

* address PR comments

* fix spelling

* don't augment eval set

* use fstring

* update augmentations.py

* new maning convention for transforms

* reverse if statement, inline ops

* reverse if statement, inline ops

* updates

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update dataloaders

* Remove additional if statement

* Remove is_train as redundant

* Cleanup

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Cleanup2

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update classifier.py

* Update augmentations.py

* fix: imshow clip warning

* update

* Revert ToTensorV2 removal

* Update classifier.py

* Update normalize values, revert uint8

* normalize image using cv2

* remove dedundant comment

* Update classifier.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* replace print with logger

* commit steps

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Allow logging models from GenericLogger (#8676)

* enhance

* revert

* allow training from scratch

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update --img argument from train.py 

single line

* fix image size from 640 to 128

* suport custom dataloader and augmentation

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* format

* Update dataloaders.py

* Single line return, single line comment, remove unused argument

* address PR comments

* fix spelling

* don't augment eval set

* use fstring

* update augmentations.py

* new maning convention for transforms

* reverse if statement, inline ops

* reverse if statement, inline ops

* updates

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update dataloaders

* Remove additional if statement

* Remove is_train as redundant

* Cleanup

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Cleanup2

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update classifier.py

* Update augmentations.py

* fix: imshow clip warning

* update

* Revert ToTensorV2 removal

* Update classifier.py

* Update normalize values, revert uint8

* normalize image using cv2

* remove dedundant comment

* Update classifier.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* replace print with logger

* commit steps

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* support final model logging

* update

* update

* update

* update

* remove curses

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update classifier.py

* Update __init__.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update

* Update

* Update

* Update

* Update dataset download

* Update dataset download

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Pass imgsz to classify_transforms()

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Cos scheduler

* Cos scheduler

* Remove unused args

* Update

* Add seed

* Add seed

* Update

* Update

* Add run(), main()

* Merge master

* Merge master

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Create YOLOv5 BaseModel class (#8829)

* Create BaseModel

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix

* Hub load device fix

* Update

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

* Add experiment

* Merge master

* Attach names

* weight decay = 1e-4

* weight decay = 5e-5

* update smart_optimizer console printout

* fashion-mnist fix

* Merge master

* Update Table

* Update Table

* Remove destroy process group

* add kwargs to forward()

* fuse fix for resnet50

* nc, names fix for resnet50

* nc, names fix for resnet50

* ONNX CPU inference fix

* revert

* cuda

* if augment or visualize

* if augment or visualize

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* New smart_inference_mode()

* Update README

* Refactor into /classify dir

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* reset defaults

* reset defaults

* fix gpu predict

* warmup

* ema half fix

* spacing

* remove data

* remove cache

* remove denormalize

* save run settings

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* verbose false on initial plots

* new save_yaml() function

* Update ci-testing.yml

* Path(data) CI fix

* Separate classification CI

* fix val

* fix val

* fix val

* smartCrossEntropyLoss

* skip validation on hub load

* autodownload with working dir root

* str(data)

* Dataset usage example

* im_show normalize

* im_show normalize

* add imagenet simple names to multibackend

* Add validation speeds

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* 24-space names

* Update bash scripts

* Update permissions

* Add bash script arguments

* remove verbose

* TRT data fix

* names generator fix

* optimize if names

* update usage

* Add local loading

* Verbose=False

* update names printing

* Add Usage examples

* Add Usage examples

* Add Usage examples

* Add Usage examples

* named_children

* reshape_classifier_outputs

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update

* update

* fix CI

* fix incorrect class substitution

* fix incorrect class substitution

* remove denormalize

* ravel fix

* cleanup

* update opt file printing

* update opt file printing

* update defaults

* add opt to checkpoint

* Add warning

* Add comment

* plot half bug fix

* Use NotImplementedError

* fix export shape report

* Fix TRT load

* cleanup CI

* profile comment

* CI fix

* Add cls models

* avoid inplace error

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Fix usage examples

* Update README

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

* Update README

Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-17 11:59:01 +02:00
Glenn Jocher 2e57b8418d
Deci.ai updates (#8928) 
* Deci.ai updates

Content implementation for https://deci.ai/

* Update README.md

* Update README.md

* Update README.md
 2022-08-11 00:01:38 +02:00
Glenn Jocher 259e055b42
Fix `--source 'path/*.jpg'` glob example (#8927) 
* Fix `--source 'path/*.jpg'` glob example

* Update README_cn.md

* Update tutorial.ipynb
 2022-08-10 21:53:12 +02:00
Glenn Jocher 4c6bdbfcde
Direct download table links (#8917) 
* Direct download table links

Per recent partner feedback

* Update README_cn.md

* Update README_cn.md

* Update README.md

* Update README_cn.md

* Update README.md
 2022-08-10 17:01:01 +02:00
Glenn Jocher 3d47fc6c55
Update README.md 2022-08-10 13:10:30 +02:00
Glenn Jocher 48a85314bc
Update README.md Environments icons (#8895) 2022-08-07 14:51:30 +02:00
Kalen Michael 0d8546a01f
Fix/improve social media bar (#8893) 
* fix: added transparent image and empty alt to social bar

* fix: remove underlines

* fix: hide underlines for all social links

* Update image URLs to ultralytics/assets

* Update bottom 3% icons

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-07 14:04:39 +02:00
Victor Sonck 378bde4bba
ClearML experiment tracking integration (#8620) 
* Add titles to matplotlib plots

* Add ClearML Experiment Tracking integration.

* Add ClearML Data Version Management automatic download when requested

* Add ClearML Hyperparameter Optimization

* ClearML save period integration

* Fix wandb breaking when used with ClearML dataset

* Fix wandb breaking when used with ClearML resume and dataset

* Add ClearML documentation

* fixed small bug in clearml integration that misreports epoch number

* Final ClearMl additions before refactor

* Add correct epoch reporting

* Add remote execution and autoscaling docs for ClearML integration

* Added images to clearml integration docs

* fixed logo alignment bug and added hpo screenshot clearml

* Fixed small epoch number bug in clearml integration

* Remove saved model flush clearml

* Cleanup clearml readme section

* Cleaned up clearml logger docstring

* Remove resume readme section clearml

* Clearml integration cleanup

* Updated ClearML documentation

* Added dark vs light icons ClearML Readme

* Clearml Readme styling

* Add better gifs

* Fixed gif file size

* Add better images in tutorial notebook

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Addressed comments in PR #8620

* Fixed circular import

* Fixed circular import

* Update tutorial.ipynb

* Update tutorial.ipynb

* Inline comment

* Restructured tutorial notebook

* Add correct ClearML link to README

* Update tutorial.ipynb

* Update general.py

* Update __init__.py

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

* Update README.md

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* spelling

* Update tutorial.ipynb

* notebook cutt.ly links

* Update README.md

* Update README.md

* cutt.ly links in tutorial

* Removed labels as they show up on last subplot only

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-05 20:50:49 +02:00
Glenn Jocher 59578f2782
Update README.md with contributors.png (#8820) 
* Update README.md with contributors.png

Replace dynamic svg from opencollective with static png for improved stability and lighter size (400kB vs 2MB).

@AyushExel

* Update README.md

* Update README.md

* Update README_cn.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-01 20:28:24 +02:00
Glenn Jocher 5c990e3674
Create README_cn.md (#8344) 
* Create README_cn.md

Add mandarin version of README

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README.md

* fix link

* fix english link

* remove line

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update .pre-commit-config.yaml

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README_cn.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update .pre-commit-config.yaml

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README_cn.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update .pre-commit-config.yaml

* Update README.md

* Update README_cn.md

* Kiera fix

* Update README_cn.md

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-06-26 17:46:39 +02:00
Glenn Jocher e4d82c1951
Update CI CPU badge (#7855) 
* Update README.md

* Update greetings.yml
 2022-05-17 13:01:18 +02:00
Glenn Jocher c4cb7c684a
Update README.md (#7686) 2022-05-03 15:18:01 -07:00
Zengyf-CVer c264795f50
Add mdformat to precommit checks and update other version (#7529) 
* Update .pre-commit-config.yaml

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update .pre-commit-config.yaml

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update CONTRIBUTING.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README.md

* Update README.md

* Update README.md

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-22 13:36:27 -07:00
Glenn Jocher 7a2a11893b
Add Architecture Summary to README Tutorials (#7146) 
* Add Architecture Summary to README Tutorials

Per https://github.com/ultralytics/yolov5/issues/6998#issuecomment-1073517405

* Update README.md
 2022-03-25 14:45:23 +01:00
Glenn Jocher 9ec51a6396
YOLOv5s6 params FLOPs fix (#6782) 2022-02-25 13:59:21 +01:00
Glenn Jocher 66aaf5184d
Pre-commit table fix (#6744) 2022-02-22 19:35:31 +01:00
Glenn Jocher 3752807c0b
YOLOv5 v6.1 release (#6739) 2022-02-22 12:35:24 +01:00
Glenn Jocher a82292ec53
Social icons after text (#6473) 
* Social icons after text

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README.md

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-02-03 19:55:19 +01:00
Glenn Jocher 7539cd75c3
Add Product Hunt social media icon (#6464) 
* Social media icons update

* fix URL

* Update README.md
 2022-01-28 20:23:17 +01:00
Glenn Jocher ff8646cdea
Update workflows (#6427) 
* Workflow updates

* quotes fix

* best to weights fix
 2022-01-25 17:33:22 -05:00
Glenn Jocher ad565e31d2
Update README speed reproduction command (#6228) 2022-01-06 09:55:31 -08:00
Glenn Jocher ec4b6dd2a3
Update export format docstrings (#6151) 
* Update export documentation

* Cleanup

* Update export.py

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README.md

* Update README.md

* Update README.md

* Update train.py

* Update train.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-01-02 16:09:45 -08:00
Glenn Jocher d699c21c75
YOLOv5s6 params and FLOPs fix (#5977) 2021-12-14 11:24:39 +01:00
Glenn Jocher 1075488d89
Single-command multiple-model export (#5882) 
* Export multiple models in series

Export multiple models in series by adding additional `*.pt` files to the `--weights` argument, i.e.:

```bash
python export.py --include tflite --weights yolov5n.pt  # export 1 model
python export.py --include tflite --weights yolov5n.pt yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt  # export 5 models
```

* Update export.py

* Update README.md
 2021-12-04 16:28:40 +01:00
Glenn Jocher b8f979bafa
Inside Ultralytics video https://youtu.be/Zgi9g1ksQHc (#5546) 
* Update detect.py Usage examples

* Inside Ultralytics at https://youtu.be/Zgi9g1ksQHc
 2021-11-06 20:34:54 +01:00
Glenn Jocher 4c0982a243
Update README.md (#5438) 
2-line update
 2021-11-01 18:28:14 +01:00
Jirka Borovec ed887b5976
Add pre-commit CI actions (#4982) 
* define pre-commit

* add CI code

* configure

* apply pre-commit

* fstring

* apply MD

* pre-commit

* Update torch_utils.py

* Update print strings

* notes

* Cleanup code-format.yml

* Update setup.cfg

* Update .pre-commit-config.yaml

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-28 18:35:01 +02:00
Glenn Jocher 6b31cfd793
Highlight contributors in README (#5173) 
* Highlight contributors in README

* Adjust Contact section

* Update README.md
 2021-10-14 12:49:16 -07:00
Glenn Jocher 956be8e642
YOLOv5 release v6.0 (#5141) 
* Update P5 models

* Update P6 models

* Update with GFLOPs and Params

* Update with GFLOPs and Params

* Update README

* Update

* Update README

* Update

* Update

* Add times

* Update README

* Update results

* Update results

* Update results

* Update hyps

* Update plots

* Update plots

* Update README.md

* Add nano models to hubconf.py
 2021-10-11 23:47:18 -07:00
Glenn Jocher b20e381492
Update README.md (#5015) 2021-09-30 14:06:00 -07:00
Kalen Michael cce313d67d
Add YOLOv5 Survey link to README.md (#5000) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* merged in upstream

* removed ghost code

* added survey

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-30 13:25:48 -07:00
Glenn Jocher 42354d7025
Update README.md 2021-09-29 16:56:10 -07:00
Kalen Michael 29acedf7dd
Add Roboflow to README (#4972) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* merged in upstream

* removed ghost code

* added integrations section removed competition

* attempt column build

* attempt column build

* moved new to header

* removed center column

* utm

* Update README.md

* Update README.md

* updated logo sizes

* resized logos

* fixed links

* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-28 09:35:46 -07:00
Glenn Jocher e687873436
Add Slack Forum badge to README (#4930) 
Add badge with link to join the new YOLOv5 Slack Forum!
https://join.slack.com/t/ultralytics/shared_invite/zt-w29ei8bp-jczz7QYUmDtgo6r6KcMIAg
 2021-09-26 10:54:55 -07:00
Glenn Jocher 11e4aebfef
Update README.md (#4387) 2021-08-11 18:33:34 +02:00
Glenn Jocher b60b62e874
PyCharm reformat (#4209) 
* PyCharm reformat

* YAML reformat

* Markdown reformat
 2021-07-28 23:35:14 +02:00
Ayush Chaurasia e88e8f7a98
W&B: Restructure code to support the new dataset_check() feature (#4197) 
* Improve docstrings and run names

* default wandb login prompt with timeout

* return key

* Update api_key check logic

* Properly support zipped dataset feature

* update docstring

* Revert tuorial change

* extend changes to log_dataset

* add run name

* bug fix

* bug fix

* Update comment

* fix import check

* remove unused import

* Hardcore .yaml file extension

* reduce code

* Reformat using pycharm

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-28 17:40:08 +02:00
Glenn Jocher d17b45eaad
Update README.md (#4143) 2021-07-25 01:22:00 +02:00
Glenn Jocher 39ef6c7a80
Update README.md (#4134) 2021-07-24 12:36:07 +02:00
Glenn Jocher dd62e2d05c
Update README.md (#4041) 
* Update README.md

* Update README.md

* Update README.md
 2021-07-17 23:12:27 +02:00
Glenn Jocher 720aaa65c8
Rename `test.py` to `val.py` (#4000) 2021-07-14 15:43:54 +02:00
Glenn Jocher 8ee9fd1505
Update README.md (#3996) 2021-07-13 23:07:09 +02:00
batrlatom 5e976a274d
Fix competition link (#3799) 
* link to the competition repaired

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-27 15:28:50 +02:00
Glenn Jocher 9dc5d35fce
Update README.md fix banner width (#3785) 2021-06-26 13:05:46 +02:00
Glenn Jocher 463628a4d8
Update README.md (#3684) 2021-06-18 17:12:42 +02:00
SpongeBab df7706d802
Update README.md (#3650) 
Be more user-friendly to new users
 2021-06-17 12:37:53 +02:00
Glenn Jocher 30e4c4f092
Update README.md (#3624) 2021-06-15 15:44:10 +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 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 f3c3d2ce5d
Merge `develop` branch into `master` (#3518) 
* update ci-testing.yml (#3322)

* update ci-testing.yml

* update greetings.yml

* bring back os matrix

* update ci-testing.yml (#3322)

* update ci-testing.yml

* update greetings.yml

* bring back os matrix

* Enable direct `--weights URL` definition (#3373)

* Enable direct `--weights URL` definition

@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```

* cleanup

* bug fixes

* weights = attempt_download(weights)

* Update experimental.py

* Update hubconf.py

* return bug fix

* comment mirror

* min_bytes

* Update tutorial.ipynb (#3368)

add Open in Kaggle badge

* `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)

* Update datasets.py

* comment

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* COCO evolution fix (#3388)

* COCO evolution fix

* cleanup

* update print

* print fix

* Create `is_pip()` function (#3391)

Returns `True` if file is part of pip package. Useful for contextual behavior modification.

```python
def is_pip():
    # Is file in a pip package?
    return 'site-packages' in Path(__file__).absolute().parts
```

* Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)" (#3395)

This reverts commit 21a9607e00.

* Update FLOPs description (#3422)

* Update README.md

* Changing FLOPS to FLOPs.

Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>

* Parse URL authentication (#3424)

* Parse URL authentication

* urllib.parse.unquote()

* improved error handling

* improved error handling

* remove %3F

* update check_file()

* Add FLOPs title to table (#3453)

* Suppress jit trace warning + graph once (#3454)

* Suppress jit trace warning + graph once

Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.

* Update train.py

* Update MixUp augmentation `alpha=beta=32.0` (#3455)

Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye

* Add `timeout()` class (#3460)

* Add `timeout()` class

* rearrange order

* Faster HSV augmentation (#3462)

remove datatype conversion process that can be skipped

* Add `check_git_status()` 5 second timeout (#3464)

* Add check_git_status() 5 second timeout

This should prevent the SSH Git bug that we were discussing @KalenMike

* cleanup

* replace timeout with check_output built-in timeout

* Improved `check_requirements()` offline-handling (#3466)

Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).

* Add `output_names` argument for ONNX export with dynamic axes (#3456)

* Add output names & dynamic axes for onnx export

Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3

* use first output only + cleanup

Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423)

* fixed inference bug ,while use half precision

* replace --use-half with --half

* replace space and PEP8 in detect.py

* PEP8 detect.py

* update --half help comment

* Update test.py

* revert space

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Add additional links/resources to stale.yml message (#3467)

* Update stale.yml

* cleanup

* Update stale.yml

* reformat

* Update stale.yml HUB URL (#3468)

* Stale `github.actor` bug fix (#3483)

* Explicit `model.eval()` call `if opt.train=False` (#3475)

* call model.eval() when opt.train is False

call model.eval() when opt.train is False

* single-line if statement

* cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* check_requirements() exclude `opencv-python` (#3495)

Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494.

* Earlier `assert` for cpu and half option (#3508)

* early assert for cpu and half option

early assert for cpu and half option

* Modified comment

Modified comment

* Update tutorial.ipynb (#3510)

* Reduce test.py results spacing (#3511)

* Update README.md (#3512)

* Update README.md

Minor modifications

* 850 width

* Update greetings.yml

revert greeting change as PRs will now merge to master.

Co-authored-by: Piotr Skalski <SkalskiP@users.noreply.github.com>
Co-authored-by: SkalskiP <piotr.skalski92@gmail.com>
Co-authored-by: Peretz Cohen <pizzaz93@users.noreply.github.com>
Co-authored-by: tudoulei <34886368+tudoulei@users.noreply.github.com>
Co-authored-by: chocosaj <chocosaj@users.noreply.github.com>
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Sam_S <SamSamhuns@users.noreply.github.com>
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: edificewang <609552430@qq.com>
 2021-06-08 10:22:10 +02:00
Glenn Jocher 407dc5008e
Update README.md (#3320) 2021-05-24 17:17:32 +02:00
Glenn Jocher f12cef85a3
TorchScript, ONNX, CoreML Export tutorial title (#3142) 2021-05-12 19:49:12 +02:00
Glenn Jocher 7918eed040
Update https://ultralytics.com/images/zidane.jpg (#3140) 2021-05-12 18:30:31 +02:00
BZFYS 6c1290fe03
Update README.md (#2934) 
* Update README.md

dependencies:
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
ImportError: libgthread-2.0.so.0: cannot open shared object file: No such file or directory
ImportError: libSM.so.6: cannot open shared object file: No such file or directory
ImportError: libXrender.so.1: cannot open shared object file: No such file or directory

* replace older apt-get with apt

Code commented for now until a better understanding of the issue, and also code is not cross-platform compatible.

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-04-26 23:56:25 +02:00
Glenn Jocher e5d71223b8
Update README.md 2021-04-15 16:45:50 +02:00
Glenn Jocher 1487bc84ff
Update README.md 2021-04-12 13:27:40 +02:00
Glenn Jocher 0f395b3e3b
YOLOv5 v5.0 Release patch 1 (#2764) 
* torch.jit.trace(model, img, strict=False)

* Update check_file()

* Update hubconf.py

* Update README.md
 2021-04-11 23:11:43 +02:00
Glenn Jocher f5b8f7d54c
YOLOv5 v5.0 Release (#2762) 2021-04-11 19:23:47 +02:00
Ben Milanko e2b7bc0b32
YouTube Livestream Detection (#2752) 
* Youtube livestream detection

* dependancy update to auto install pafy

* Remove print

* include youtube_dl in deps

* PEP8 reformat

* youtube url check fix

* reduce lines

* add comment

* update check_requirements

* stream framerate fix

* Update README.md

* cleanup

* PEP8

* remove cap.retrieve() failure code

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-04-11 18:53:40 +02:00
Glenn Jocher c8c8da6079
Update README with collapsable notes (#2721) 
* Update README with collapsable notes.

* cleanup

* center table
 2021-04-06 17:54:47 +02:00
Glenn Jocher b8b862993d
Update README with Tips for Best Results tutorial (#2682) 
* Update README with Tips for Best Results tutorial

* Update README.md
 2021-04-01 15:01:00 +02:00
Glenn Jocher 51cc0962b5
Update README.md 2021-04-01 11:16:56 +02:00
Max Kolomeychenko 16206692f2
Supervisely Ecosystem (#2519) 
guide describes YOLOv5 apps collection in Supervisely Ecosystem
 2021-03-25 00:57:34 +01:00
Glenn Jocher 569757ecc0
Add autoShape() speed profiling (#2459) 
* Add autoShape() speed profiling

* Update common.py

* Create README.md

* Update hubconf.py

* cleanuip
 2021-03-13 19:50:34 -08:00
Glenn Jocher 7a6870b81f
Update README.md 2021-02-23 11:27:44 -08:00
Glenn Jocher 5a40ce65ce
Update CI badge (#2230) 2021-02-16 15:27:24 -08:00
Glenn Jocher eeb2bbf648
Add Kaggle badge (#2090) 
* Update README.md

* Update greetings.yml

* Created using Colaboratory
 2021-01-30 11:48:03 -08:00
Glenn Jocher 170d12e506
Add Amazon Deep Learning AMI environment (#2085) 
* Update greetings.yml

* Update README.md
 2021-01-29 13:49:00 -08:00
Glenn Jocher 69be8e738f
YOLOv5 v4.0 Release (#1837) 
* Update C3 module

* Update C3 module

* Update C3 module

* Update C3 module

* update

* update

* update

* update

* update

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update datasets

* update

* update

* update

* update attempt_downlaod()

* merge

* merge

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* parameterize eps

* comments

* gs-multiple

* update

* max_nms implemented

* Create one_cycle() function

* update

* update

* update

* update

* update

* update

* update

* update study.png

* update study.png

* Update datasets.py
 2021-01-04 19:54:09 -08:00
NanoCode012 14b0abe2d0
autoShape() default for PyTorch Hub models (#1692) 
* Add autoshape parameter

* Remove autoshape call in ReadMe

* Update hubconf.py

* file/URI inputs and autoshape check passthrough

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-12-26 19:58:26 -08:00
Glenn Jocher 98a46d8e82
Update README.md (#1764) 2020-12-22 23:03:08 -08:00
Glenn Jocher 7d629fde05
Update README.md 2020-11-25 11:00:29 +01:00
Glenn Jocher d336b31f20
Update README.md 2020-11-25 11:00:00 +01:00
Glenn Jocher 8c0e7092a2
Update README.md 2020-11-23 19:28:35 +01:00
Glenn Jocher 201bafc7cf
Sync train and test iou_thresh (#1465) 
* Sync train and test iou_thresh

* Sync train and test iou_thresh

* weights names .lower()

* Notebook update
 2020-11-21 12:38:35 +01:00
Glenn Jocher c4addd7761
Unified '/project/name' results saving (#1377) 
* Project/name update

* Update ci-testing.yml

* address project with path separator failure mode

* Project/name update

* address project with path separator failure mode

* Update ci-testing.yml

* detect.py default --name bug fix

* missing rstrip PR

* train/exp0 to train/exp
 2020-11-12 23:37:46 +01:00
Glenn Jocher 7841d7ba88
Update README.md 2020-11-10 17:12:35 +01:00
Glenn Jocher 67bf9a974e
Replace 'ground truth' with 'labels' (#1337) 
* Replace 'ground truth' with 'labels'

* Using torch... update
 2020-11-10 14:15:03 +01:00
Glenn Jocher 4821d076e2
Increment train, test, detect runs/ (#1322) 
* Increment train, test, detect runs/

* Update ci-testing.yml

* inference/images to data/images

* move images

* runs/exp to runs/train/exp

* update 'results saved to %s' str
 2020-11-08 19:39:05 +01:00
Glenn Jocher 74b34872fd
Add Weights & Biases tutorial (#1290) 2020-11-05 02:30:13 +01:00
Glenn Jocher 2062765e06
auto-download v3.1 models (#1259) 2020-11-01 20:11:41 +01:00
Glenn Jocher 0a52ae16dc
README Update (#1207) 
* README Update

* Update README.md

* README Update

* Update README.md
 2020-10-25 14:05:28 +01:00
stone99 b8f656742d
feat: add rtmp support (#1009) 
* feat: add rtmp support

* Update detect.py

pass tuple to source.startswith()

Co-authored-by: ryan.fu <ryan.fu@aylaasia.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-09-22 11:37:49 -07:00
Glenn Jocher 5991d144c1
Update README.md 2020-08-14 00:04:12 -07:00
Glenn Jocher 06bdd3955c
Update README.md 2020-08-13 14:31:08 -07:00
Glenn Jocher 916d4aad9a
v3.0 Release (#725) 
* initial commit

* remove yolov3-spp from test.py study

* update study --img range

* update mAP

* cleanup and speed updates

* update README plot
 2020-08-13 14:25:05 -07:00
tkianai 1ddf692bf8
Update readme to remove apex dependency (#727) 
* the value should be a list to unpacking

* filter the None labels

* update readme to remove apex dependency
 2020-08-12 21:18:29 -07:00
Glenn Jocher 941cb2680b Merge remote-tracking branch 'origin/master' 2020-08-03 19:55:57 -07:00
Glenn Jocher 52b626392f update README.md 2020-08-03 19:55:12 -07:00
Wang Xinyu 7ecf09d6da
Update README.md, deploy model with tensorrt (#623) 
* Update README.md, deploy model with tensorrt

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-03 19:41:50 -07:00
Glenn Jocher f807e7b867
Update README.md 2020-08-03 11:11:12 -07:00
Glenn Jocher 937b9db977
Update README.md 2020-08-03 11:09:46 -07:00
Glenn Jocher fdbcc8fd7f hyperparameter evolution update (#566) 2020-08-02 12:30:34 -07:00
Glenn Jocher 3b394b9e2c
Update README.md 2020-07-23 15:49:10 -07:00
Glenn Jocher 54e1762ef5
Update README.md 2020-07-23 15:43:40 -07:00
Glenn Jocher 9da56b62dd
v2.0 Release (#491) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-23 15:34:23 -07:00
NanoCode012 999804fea9
Add link to multi-gpu train tutorial (#476) 
* Add link to multi-gpu train tutorial

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-22 11:56:48 -07:00
Jirka e8ea772384 revert test module to confuse users... 2020-07-17 01:16:22 +02:00
Jirka eac0dbc46a rename 2020-07-16 12:11:38 +02:00
Jirka 613b9d6d98 add badge 2020-07-16 10:33:43 +02:00