Glenn Jocher
070af88108
Fix `yaml.safe_load()` ignore emoji errors ( #5060 )
3 years ago
Glenn Jocher
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.
3 years ago
Glenn Jocher
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
3 years ago
Zhiqiang Wang
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
3 years ago
Glenn Jocher
b0ade48457
Fix missing `opt.device` on `--task study` ( #5031 )
3 years ago
Glenn Jocher
cce7e78fae
Created using Colaboratory
3 years ago
Glenn Jocher
89a6e9ffa7
Compute loss on final val ( #5017 )
3 years ago
Glenn Jocher
b20e381492
Update README.md ( #5015 )
3 years ago
Glenn Jocher
452e84703c
Update train.py ( #5014 )
3 years ago
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>
3 years ago
Glenn Jocher
ba99092304
Update W&B README ( #5006 )
3 years ago
Glenn Jocher
1916226841
Created using Colaboratory
3 years ago
Glenn Jocher
42354d7025
Update README.md
3 years ago
Kalen Michael
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>
3 years ago
Glenn Jocher
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.
3 years ago
Glenn Jocher
9988059b10
Clickable CI badge ( #4985 )
3 years ago
Glenn Jocher
6b19f728a1
Created using Colaboratory
3 years ago
Glenn Jocher
cff7d2af45
Created using Colaboratory
3 years ago
Glenn Jocher
0c93ec762a
Faster `--img 64` CI tests ( #4979 )
3 years ago
Glenn Jocher
94705a9528
Add Hub custom models to CI tests ( #4978 )
* Update ci-testing.yml for Hub custom model tests
* Update ci-testing.yml
3 years ago
Diego Montes
0c87478713
Update wandb_utils.py ( #4953 )
`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
3 years ago
Glenn Jocher
38c779b099
Created using Colaboratory
3 years ago
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>
3 years ago
Glenn Jocher
fb982d6030
Fix relative `ROOT` Pytorch Hub custom model bug ( #4974 )
* Fix relative `ROOT` Pytorch Hub custom model bug
* Update yolo.py
3 years ago
Diego Montes
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>
3 years ago
Glenn Jocher
2993c3fa7a
Add `roboflow` ( #4956 )
3 years ago
Glenn Jocher
3aeb57d66f
Created using Colaboratory
3 years ago
Glenn Jocher
5ed28603cf
Update relative `ROOT` logic ( #4955 )
* Update relative `ROOT` logic
* python 3.9 Path().is_relative_to() removal
3 years ago
Glenn Jocher
4e04cb0dc8
Allow YOLOv5 execution from arbitrary `cwd` ( #4954 )
* Allow YOLOv5 execution from arbitrary `cwd`
* Fix str bugs
3 years ago
Glenn Jocher
a820b43aca
Automatic Chinese fonts plotting ( #4951 )
* Automatic Chinese fonts plotting
* Default PIL=False
3 years ago
Glenn Jocher
c5ba2abb4a
Update Dockerfile ( #4935 )
3 years ago
Glenn Jocher
5a8e4343d8
Scope `check_file()` search space ( #4933 )
`check_file()` is now limited to searching opt-in directories: /data, /models, /utils. This prevents large non-project directories like /.git and /venv from being searched, which may cause `check_file()` to slow significantly.
3 years ago
Glenn Jocher
793383232f
Update default Albumentations ( #4931 )
3 years ago
Glenn Jocher
d856c48298
Validate `best.pt` on train end ( #4889 )
* Validate best.pt on train end
* 0.7 iou for COCO only
* pass callbacks
* active model.float() if not half
* print Validating best.pt...
* add newline
3 years ago
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
3 years ago
Glenn Jocher
39c17ce0b9
Fix `root` referenced before assignment ( #4920 )
* Fix `root` referenced before assignment
Fix for bug introduced by #4919 discovered on VOC autodownload:
```
python train.py --data VOC.yaml
```
* Cleanup
3 years ago
Glenn Jocher
a64a4c839f
Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()` ( #4919 )
* Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()`
* Cleanup
3 years ago
Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 years ago
Jiacong Fang
2c2ef25f8b
TensorFlow.js export enhancements ( #4905 )
* Add arguments to TensorFlow NMS call
* Add regex substitution to reorder Identity_*
* Delete reorder in docstring
* Cleanup
* Cleanup2
* Removed `+ \` on string ends (not needed)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Jiacong Fang
ce7fa81d4e
Avoid out-of-image class labels ( #4902 )
* Avoid out-of-image class labels
* Update plots.py
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
NauchtanRobotics
dad8660540
Fix arg help string to match 'classes' arg name ( #4893 )
3 years ago
Ayush Chaurasia
59aae85a7e
W&B: Fix dataset check ( #4879 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
* fix
* reformat
3 years ago
Ayush Chaurasia
9febea79de
W&B: Login only in master processes ( #4866 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
3 years ago
Glenn Jocher
8ad9e4ed5b
Annotator `check_font()` RANK -1 remove progress ( #4864 )
* Annotator `check_font()` RANK -1 remove progress
* Cleanup
3 years ago
Glenn Jocher
2b3109eeb0
Fix DDP destruction `LOGGER.info()` ( #4863 )
3 years ago
Glenn Jocher
2809616345
Update Dockerfile ( #4862 )
3 years ago
Glenn Jocher
40d1c80503
Update Dockerfile ( #4861 )
3 years ago
Glenn Jocher
9ef94940aa
Update greetings.yml ( #4856 )
3 years ago
Glenn Jocher
4c839eeb10
Simplify `check_requirements()` usage ( #4855 )
* Simplify `check_requirements()` usage
* remove assert, print()
3 years ago
Glenn Jocher
4d1a2ac87e
Update `sys.path.append(str(ROOT))` ( #4852 )
* Update `sys.path.append(str(ROOT))`
* Cleanup
3 years ago