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

1172 Commits

Author SHA1 Message Date
Glenn Jocher
987c226849 seaborn plotting correlogram addition 2020-08-29 15:13:58 -07:00
rafale77
9776e70988
torch.ops.torchvision.nms (#860) 
Don't load the entire torchvision library just for nms when the function is already in the torch library.
 2020-08-27 11:13:41 -07:00
Glenn Jocher
31d6c389f2 Update general.py plot_evolution() 2020-08-25 23:08:14 -07:00
Willie Maddox
455f7b8f76
Update general.py (#823) 
Fixes #822

`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
 2020-08-25 13:01:17 -07:00
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements (#736) 
eps reduced to 1e-9 for additional precaution.
 2020-08-25 12:56:39 -07:00
Glenn Jocher
82ed33a052 generalize webcam detection with opt.source.isnumeric() (yolov3 #1462) 2020-08-25 11:07:38 -07:00
Glenn Jocher
4d7f222f73 Update export.py with v3.0 Hardswish() support 2020-08-24 21:47:49 -07:00
Glenn Jocher
8666bc507e Create evolve.sh for infinite hyp evolution 2020-08-24 20:27:28 -07:00
Glenn Jocher
71209a6099 exportable Hardswish() implementation 2020-08-22 16:59:31 -07:00
Glenn Jocher
f7e075fe93 gsutil cp yaml when evolving to GCP bucket 2020-08-22 15:47:39 -07:00
Glenn Jocher
9dbaa1bbbe Update plot_results() 2020-08-22 15:32:05 -07:00
Glenn Jocher
ea5e97fb9b
FP16 NMS (#814) 2020-08-21 14:44:25 -07:00
Glenn Jocher
b42e8a531b optimize imports 2020-08-20 21:17:40 -07:00
Glenn Jocher
5e0b90de8f CIoU nan bug fix (#736) 2020-08-20 10:37:55 -07:00
Glenn Jocher
ebafd1ead5
single command --resume (#756) 
* single command --resume

* else check files, remove TODO

* argparse.Namespace()

* tensorboard lr

* bug fix in get_latest_run()
 2020-08-17 16:28:43 -07:00
Francisco Ingham
26c3b11fdc
Update torch_utils.py (#751) 2020-08-16 13:58:59 -07:00
Glenn Jocher
f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
Quan Lin
66744a0df1
Update utils/google_utils.py (#733) 
To download weights from v3.0 instead of v2.0
 2020-08-13 18:44:14 -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
Hatovix
56c2c344ff
Fix list paths (#721) 
* Add list paths on check_dataset

* missing raise statement

* Update general.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-12 18:53:38 -07:00
Marc
a925f283a7
max workers for dataloader (#722) 2020-08-12 13:57:36 -07:00
Glenn Jocher
d2da523053 Merge remote-tracking branch 'origin/master' 2020-08-11 20:11:58 -07:00
Glenn Jocher
9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
Marc
1f92422e20
Github release assets model autodownload (#711) 
* assets autodownload

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-11 17:17:24 -07:00
NanoCode012
4949401a94
Fix redundant outputs via Logging in DDP training (#500) 
* Change print to logging

* Clean function set_logging

* Add line spacing

* Change leftover prints to log

* Fix scanning labels output

* Fix rank naming

* Change leftover print to logging

* Reorganized DDP variables

* Fix type error

* Make quotes consistent

* Fix spelling

* Clean function call

* Add line spacing

* Update datasets.py

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-11 11:18:45 -07:00
tkianai
542833c997
Filter the None labels (#705) 
* the value should be a list to unpacking

* filter the None labels
 2020-08-11 11:15:48 -07:00
tkianai
9ae868364a
the value should be a list to unpacking (#688) 2020-08-10 11:23:48 -07:00
Glenn Jocher
41523e2c91
Dataset autodownload feature addition (#685) 
* initial commit

* move download scripts into data/scripts

* new check_dataset() function in general.py

* move check_dataset() out of with context

* Update general.py

* DDP update

* Update general.py
 2020-08-09 20:52:57 -07:00
Glenn Jocher
3c6e2f7668
Single-source training (#680) 
* Single-source training

* Extract hyperparameters into seperate files

* weight decay scientific notation yaml reader bug fix

* remove import glob

* intersect_dicts() implementation

* 'or' bug fix

* .to(device) bug fix
 2020-08-09 02:12:44 -07:00
Glenn Jocher
3fcd365ba3 Fix curl download on Windows (#669) 2020-08-07 12:13:55 -07:00
NanoCode012
fe5b3f8712
Fix curl download on Windows (#669) 
* Update ci-testing.yml

* Fix windows download

* Fix cookie and curl download issue from gdrive

* Revert "Update ci-testing.yml"

This reverts commit 7389d2238df4dd8bd3b6ddded0f12e551c8fd0b1.

* Update google_utils.py

PEP8

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-07 12:10:59 -07:00
Glenn Jocher
7eaf225d55 zero-target training bug fix (#609) 2020-08-05 13:35:31 -07:00
Glenn Jocher
f346da9f2b update activations.py 2020-08-03 17:18:31 -07:00
Jirka Borovec
d5b6416c87
Explicit Imports (#498) 
* expand imports

* optimize

* miss

* fix
 2020-08-02 15:47:36 -07:00
Glenn Jocher
ec7a926163 AutoAnchor update to display anchors/target 2020-08-02 14:38:45 -07:00
Glenn Jocher
d989bc9260 remove NBSP 2020-08-02 14:23:05 -07:00
Glenn Jocher
023e37807c hyperparameter evolution update (#566) 2020-08-02 13:18:03 -07:00
Glenn Jocher
64e6d199ba hyperparameter evolution update (#566) 2020-08-02 13:17:26 -07:00
Glenn Jocher
305c6a028a compute_loss() leaf variable update 2020-08-02 10:30:43 -07:00
Glenn Jocher
c1a2a7a411 hyperparameter evolution bug fix (#566) 2020-08-01 23:00:10 -07:00
Glenn Jocher
e32abb5fb9 hyperparameter evolution bug fix (#566) 2020-08-01 19:15:48 -07:00
Glenn Jocher
127cbeb3f5 hyperparameter expansion to flips, perspective, mixup 2020-08-01 13:47:54 -07:00
Glenn Jocher
23d2bde01f leaf variable --single-cls training bug fix (#593) 2020-08-01 11:42:41 -07:00
Glenn Jocher
bcd452c482 replace random_affine() with random_perspective() 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-31 15:53:52 -07:00
Glenn Jocher
bb87276d80
update build_targets() (#589) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-31 14:34:13 -07:00
Liu Changyu
c020875b17
PyTorch 1.6.0 update with native AMP (#573) 
* PyTorch have Automatic Mixed Precision (AMP) Training.

* Fixed the problem of inconsistent code length indentation

* Fixed the problem of inconsistent code length indentation

* Mixed precision training is turned on by default
 2020-07-31 10:52:45 -07:00
Glenn Jocher
c8892672a7 Funnel ReLU (FReLU) @staticmethod bug fix (#556) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-30 17:38:15 -07:00
Glenn Jocher
131782a2aa Funnel ReLU (FReLU) (#556) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-30 17:35:34 -07:00
AlexWang1900
4b074d9d9d
Funnel ReLU (FReLU) (#556) 
* fix #541 #542

* Update train.py

* Add Frelu

* Update activations.py

PEP8 and format updates for commonality with models.common.Conv()

* Update activations.py

Update case

* Update activations.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-07-30 17:25:49 -07:00
YuriRibeiro
7d78b17a69
Fix utils.py nonzero() warning (#551) 2020-07-29 12:14:11 -07:00