mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,747 Commits 10 Branches 10 Tags
Commit Graph

1747 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
57ce73cd4f
Update datasets.py comments (#5088) 2021-10-07 16:04:08 -07:00
Glenn Jocher
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
Glenn Jocher
070af88108
Fix yaml.safe_load() ignore emoji errors (#5060) 2021-10-05 13:41:52 -07:00
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.
 2021-10-04 18:48:46 -07:00
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
 2021-10-04 15:48:55 -07:00
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
 2021-10-03 17:54:40 -07:00
Glenn Jocher
b0ade48457
Fix missing opt.device on --task study (#5031) 2021-10-02 17:55:55 -07:00
Glenn Jocher
cce7e78fae Created using Colaboratory 2021-09-30 19:10:55 -07:00
Glenn Jocher
89a6e9ffa7
Compute loss on final val (#5017) 2021-09-30 15:16:11 -07:00
Glenn Jocher
b20e381492
Update README.md (#5015) 2021-09-30 14:06:00 -07:00
Glenn Jocher
452e84703c
Update train.py (#5014) 2021-09-30 13:57:53 -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
ba99092304
Update W&B README (#5006) 2021-09-30 00:35:00 -07:00
Glenn Jocher
1916226841 Created using Colaboratory 2021-09-29 22:18:16 -07:00
Glenn Jocher
42354d7025
Update README.md 2021-09-29 16:56:10 -07:00
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>
 2021-09-29 10:48:45 -07:00
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.
 2021-09-28 20:11:20 -07:00
Glenn Jocher
9988059b10
Clickable CI badge (#4985) 2021-09-28 17:46:14 -07:00
Glenn Jocher
6b19f728a1 Created using Colaboratory 2021-09-28 17:33:48 -07:00
Glenn Jocher
cff7d2af45 Created using Colaboratory 2021-09-28 17:29:05 -07:00
Glenn Jocher
0c93ec762a
Faster --img 64 CI tests (#4979) 2021-09-28 12:41:28 -07:00
Glenn Jocher
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
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.
 2021-09-28 12:22:31 -07:00
Glenn Jocher
38c779b099 Created using Colaboratory 2021-09-28 10:04:56 -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
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
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>
 2021-09-27 20:16:23 -07:00
Glenn Jocher
2993c3fa7a
Add roboflow (#4956) 2021-09-27 19:00:05 -07:00
Glenn Jocher
3aeb57d66f Created using Colaboratory 2021-09-27 18:32:22 -07:00
Glenn Jocher
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
Glenn Jocher
4e04cb0dc8
Allow YOLOv5 execution from arbitrary cwd (#4954) 
* Allow YOLOv5 execution from arbitrary `cwd`

* Fix str bugs
 2021-09-27 17:40:20 -07:00
Glenn Jocher
a820b43aca
Automatic Chinese fonts plotting (#4951) 
* Automatic Chinese fonts plotting

* Default PIL=False
 2021-09-27 13:48:15 -07:00
Glenn Jocher
c5ba2abb4a
Update Dockerfile (#4935) 2021-09-26 19:16:14 -07:00
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.
 2021-09-26 17:11:46 -07:00
Glenn Jocher
793383232f
Update default Albumentations (#4931) 2021-09-26 12:10:33 -07:00
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
 2021-09-26 11:54:53 -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
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
 2021-09-25 09:16:14 -07:00
Glenn Jocher
a64a4c839f
Replace os.system('unzip file.zip') -> ZipFile.extractall() (#4919) 
* Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()`

* Cleanup
 2021-09-25 08:52:36 -07:00
Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments (#4914) 2021-09-25 06:43:24 -07:00
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>
 2021-09-24 14:18:15 -07:00
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>
 2021-09-24 09:23:28 -07:00
NauchtanRobotics
dad8660540
Fix arg help string to match 'classes' arg name (#4893) 2021-09-24 06:44:01 -07:00
Ayush Chaurasia
59aae85a7e
W&B: Fix dataset check (#4879) 
* evolve fix

* Enable login timeout

* fix pkg

* check rank

* don't relogin

* fix

* reformat
 2021-09-21 20:02:32 +02:00
Ayush Chaurasia
9febea79de
W&B: Login only in master processes (#4866) 
* evolve fix

* Enable login timeout

* fix pkg

* check rank

* don't relogin
 2021-09-20 16:19:34 +02:00
Glenn Jocher
8ad9e4ed5b
Annotator check_font() RANK -1 remove progress (#4864) 
* Annotator `check_font()` RANK -1 remove progress

* Cleanup
 2021-09-20 13:57:23 +02:00
Glenn Jocher
2b3109eeb0
Fix DDP destruction LOGGER.info() (#4863) 2021-09-20 13:12:48 +02:00
Glenn Jocher
2809616345
Update Dockerfile (#4862) 2021-09-20 13:10:41 +02:00
Glenn Jocher
40d1c80503
Update Dockerfile (#4861) 2021-09-20 12:22:24 +02:00
Glenn Jocher
9ef94940aa
Update greetings.yml (#4856) 2021-09-18 18:43:26 +02:00