da56cd1281
train
2025-04-17 15:53:08 +08:00
9899722b9e
general load file's size
2025-04-17 11:57:42 +08:00
4afe9a152b
alllabelsdone
2025-04-16 22:53:27 +08:00
70cb8e0b1a
reset
2025-04-16 22:32:20 +08:00
249810fed5
second
2025-04-16 22:12:22 +08:00
c2042bf9bc
this is the first commit
2025-04-16 22:06:39 +08:00
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
938efe5dcf
update `detect.py` in order to support torch script ( #5109 )
...
* update detect.py in order to support torch script
This change assumes the torchscrip file was previously saved with `export.py`
* update `detect.py` for torchscript support
Simple update for torchscript support. Assumes the torchscript file has been generated with `export.py`
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-11 21:56:06 -07:00
0bf24cf641
Add OpenCV DNN option for ONNX inference ( #5136 )
...
* Add OpenCV DNN option for ONNX inference
Usage:
```bash
python detect.py --weights yolov5s.onnx # ONNX Runtime inference
python detect.py --weights yolov5s.onnx -dnn # OpenCV DNN inference
```
* DNN prediction to tensor
* Update detect.py
2021-10-11 12:39:20 -07:00
9d75e42f98
Refactor `Detect()` anchors for ONNX <> OpenCV DNN compatibility ( #4833 )
...
* refactor anchors and anchor_grid in Detect Layer
* fix CI failures by adding compatibility
* fix tf failure
* fix different devices errors
* Cleanup
* fix anchors overwriting issue
* better refactoring
* Remove self.anchor_grid shape check (redundant with self.grid check)
Also PEP8 / 120 line width
* Convert _make_grid() from static to dynamic method
* Remove anchor_grid.to(device)
clone() should already clone to same device as self.anchors
* fix different devices error
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-11 09:58:42 -07:00
153873e9e4
Fix `ROOT` as relative path ( #5129 )
...
* use os.path.relpath instead of relative_to
* use os.path.relpath instead of relative_to
* Remove os.path from val.py
* Remove os.path from train.py
* Update detect.py import to os
* Update export.py import to os
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-11 09:47:24 -07:00
4cf7d487a6
Update plots.py ( #5127 )
2021-10-11 01:59:22 -07:00
7d37b3c52e
Update val.py `pad = 0.0 if task == speed else 0.5` ( #5121 )
...
* Update val.py `pad = 0.0 if task == speed else 0.5`
* Cleanup
2021-10-10 23:20:42 -07:00
ba4b79de8b
Update val.py `--speed` and `--study` usages ( #5120 )
2021-10-10 21:15:28 -07:00
4f9718abe6
Pass `--device` for `--task study` ( #5118 )
2021-10-10 15:07:26 -07:00
a0e15046c3
Fix different devices bug when moving model from GPU to CPU ( #5110 )
...
* fix different devices bug
* extend _apply() instead of to() for a general fix
* Only apply if Detect() is last layer
Co-authored-by: Jebastin Nadar <njebastin10@gmail.com>
* Indent fix
* Add comment to yolo.py
* Add comment to common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-10 14:24:05 -07:00
4a6dfffdaa
Pass `LOCAL_RANK` to `torch_distributed_zero_first()` ( #5114 )
...
Co-authored-by: qiningonline <qiningonline@gmail.com>
2021-10-09 18:41:50 -07:00
97b6b14abe
Add `requests` to requirements.txt ( #5112 )
...
* Requirements updated
1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer.
2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies.
* Sort alphabetically
* Update requirements.txt
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-09 15:51:01 -07:00
276b674513
Fix SKU-110K HUB: `OSError` ( #5106 )
2021-10-09 00:12:59 -07:00
48b00dbc0d
Re-open IP-camera videostream if disconnected ( #5074 )
...
* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability
* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability
* `self.imgs[i] *= 0` bug fix and cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-07 23:26:43 -07:00
57ce73cd4f
Update datasets.py comments ( #5088 )
2021-10-07 16:04:08 -07:00
ce8e5dc8bb
Update Dockerfile to `ADD` Arial.ttf ( #5084 )
...
* Update Dockerfile to `ADD` Arial.ttf
Fix for `Arial.ttf download in Docker` #5078
* Cleanup
* Cleanup2
2021-10-07 14:16:33 -07:00
070af88108
Fix `yaml.safe_load()` ignore emoji errors ( #5060 )
2021-10-05 13:41:52 -07:00
5afc9c25ef
Implement `--save-period` locally ( #5047 )
...
This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
```
python train.py --save-period 50 # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
```
This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
2021-10-04 18:48:46 -07:00
d133968b1e
Clip TTA Augmented Tails ( #5028 )
...
* Clip TTA Augmented Tails
Experimental TTA update.
* Update yolo.py
* Update yolo.py
* Update yolo.py
* Update yolo.py
2021-10-04 15:48:55 -07:00
1922ddeac0
Fix pylint: do not use bare 'except' ( #5025 )
...
* Fix E722, do not use bare 'except'
* Remove used codes
* Add FileNotFoundError in LoadImagesAndLabels
* Remove AssertionError
* Ignore LoadImagesAndLabels
* Ignore downloads.py
* Ignore torch_utils.py
* Ignore train.py
* Ignore datasets.py
* Enable utils/download.py
* Fixing exception in thop
* Remove unused code
* Fixing exception in LoadImagesAndLabels
* Fixing exception in exif_size
* Fixing exception in parse_model
* Ignore exceptions in requests
* Revert the exception as suggested
* Revert the exception as suggested
2021-10-03 17:54:40 -07:00
b0ade48457
Fix missing `opt.device` on `--task study` ( #5031 )
2021-10-02 17:55:55 -07:00
cce7e78fae
Created using Colaboratory
2021-09-30 19:10:55 -07:00
89a6e9ffa7
Compute loss on final val ( #5017 )
2021-09-30 15:16:11 -07:00
b20e381492
Update README.md ( #5015 )
2021-09-30 14:06:00 -07:00
452e84703c
Update train.py ( #5014 )
2021-09-30 13:57:53 -07:00
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
ba99092304
Update W&B README ( #5006 )
2021-09-30 00:35:00 -07:00
1916226841
Created using Colaboratory
2021-09-29 22:18:16 -07:00
42354d7025
Update README.md
2021-09-29 16:56:10 -07:00
76d301bd21
Fix URL parsing bug ( #4998 )
...
* added callbacks
* added back callback to main
* added save_dir to callback output
* merged in upstream
* removed ghost code
* fixed parsing error for google temp links
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-09-29 10:48:45 -07:00
cd35a009ba
Revert `torch.hub.load()` test ( #4986 )
...
Temporarily reverts https://github.com/ultralytics/yolov5/pull/4978 until torch 1.10 is released, which should resolve `urllib.error.HTTPError: HTTP Error 403: rate limit exceeded` errors generated by torch hub from GitHub actions runners.
2021-09-28 20:11:20 -07:00
9988059b10
Clickable CI badge ( #4985 )
2021-09-28 17:46:14 -07:00
6b19f728a1
Created using Colaboratory
2021-09-28 17:33:48 -07:00
cff7d2af45
Created using Colaboratory
2021-09-28 17:29:05 -07:00
0c93ec762a
Faster `--img 64` CI tests ( #4979 )
2021-09-28 12:41:28 -07:00
94705a9528
Add Hub custom models to CI tests ( #4978 )
...
* Update ci-testing.yml for Hub custom model tests
* Update ci-testing.yml
2021-09-28 12:28:18 -07:00
0c87478713
Update wandb_utils.py ( #4953 )
...
`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
2021-09-28 12:22:31 -07:00
38c779b099
Created using Colaboratory
2021-09-28 10:04:56 -07:00
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
fb982d6030
Fix relative `ROOT` Pytorch Hub custom model bug ( #4974 )
...
* Fix relative `ROOT` Pytorch Hub custom model bug
* Update yolo.py
2021-09-28 06:36:12 -07:00
c1bed601e9
Fix `isascii()` method calls for python 3.6 ( #4958 )
...
* fix isascii for python3.6
* update comment with python 3.7 note
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-09-27 20:16:23 -07:00
2993c3fa7a
Add `roboflow` ( #4956 )
2021-09-27 19:00:05 -07:00
3aeb57d66f
Created using Colaboratory
2021-09-27 18:32:22 -07:00
5ed28603cf
Update relative `ROOT` logic ( #4955 )
...
* Update relative `ROOT` logic
* python 3.9 Path().is_relative_to() removal
2021-09-27 18:26:41 -07:00
StrickenSun
Glenn Jocher
Andrei Ionut Damian
Jebastin Nadar
Malte Lorbach
Nan
hamid mohammadi
Egor Orel
Zhiqiang Wang
Kalen Michael
Diego Montes