788 Commits

Author SHA1 Message Date
Glenn Jocher
7eb23e3c1d
YOLOv5 v6.0 compatibility update (#1857)
* Initial commit

* Initial commit

* Cleanup

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

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

* Fix precommit errors

* Remove TF builds from CI

* export last.pt

* Created using Colaboratory

* [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>
2021-11-14 22:22:59 +01:00
Glenn Jocher
4d0c2e6eee YOLOv5 v5.0 release compatibility update for YOLOv3 2021-05-30 18:55:56 +02:00
Glenn Jocher
8eb4cde090
YOLOv5 v5.0 release compatibility update for YOLOv3 (#1737)
* YOLOv5 v5.0 release compatibility update

* Update README

* Update README

* Conv act LeakyReLU(0.1)

* update plots_study()

* update speeds
2021-04-12 18:00:47 +02:00
huntr.dev | the place to protect open source
cf5db95953
Security Fix for Arbitrary Code Execution - huntr.dev (#1672)
* fixed arbitary code execution

* Update train.py

* Full to Safe

Co-authored-by: Asjid Kalam <asjid.kalam@gmail.com>
Co-authored-by: Jamie Slome <jamie@418sec.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-01-25 09:39:34 -08:00
Glenn Jocher
166a4d590f
v9.1 release (#1658) 2021-01-12 23:05:32 -08:00
Glenn Jocher
4a07280884
Pycocotools best.pt after COCO train (#1593) 2020-12-06 14:58:50 +01:00
Glenn Jocher
4f890d13ee
Daemon thread plots (#1578) 2020-11-30 16:47:28 +01:00
Glenn Jocher
76807fae71
YOLOv5 Forward Compatibility Update (#1569)
* YOLOv5 forward compatibility update

* add data dir

* ci test yolov3

* update build_targets()

* update build_targets()

* update build_targets()

* update yolov3-spp.yaml

* add yolov3-tiny.yaml

* add yolov3-tiny.yaml

* Update yolov3-tiny.yaml

* thop bug fix

* Detection() device bug fix

* Use torchvision.ops.nms()

* Remove redundant download mirror

* CI tests with yolov3-tiny

* Update README.md

* Synch train and test iou_thresh

* update requirements.txt

* Cat apriori autolabels

* Confusion matrix

* Autosplit

* Autosplit

* Update README.md

* AP no plot

* Update caching

* Update caching

* Caching bug fix

* --image-weights bug fix

* datasets bug fix

* mosaic plots bug fix

* plot_study

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* update

* Update README

* Update README

* Update README.md

* Update README.md

* results png

* Update README

* Targets scaling bug fix

* update plot_study

* update plot_study

* update plot_study

* update plot_study

* Targets scaling bug fix

* Finish Readme.md

* Finish Readme.md

* Finish Readme.md

* Update README.md

* Creado con Colaboratory
2020-11-26 20:24:00 +01:00
Glenn Jocher
bf34ae007f Global code reformat and optimize imports 2020-08-22 17:49:27 -07:00
Glenn Jocher
63996a8bfe --resume update 2020-06-30 21:45:06 -07:00
Glenn Jocher
f8e5338f0a --resume epochs update 2020-06-30 16:19:56 -07:00
NanoCode012
a587d39cd4
Fixed train.py SyntaxError due to last commit (#1336)
Fixed unexpected character after line continuation character on line 148,150, and 151
2020-06-24 11:37:09 -07:00
Oulbacha Reda
a97f350461
Non-output layer freeze in train.py (#1333)
Freeze layers that aren't of type YOLOLayer and that aren't the conv layers preceeding them
2020-06-22 13:15:40 -07:00
FuLin
10dc08f91b
revert value of gs back to 32(from 64) (#1317) 2020-06-19 09:54:57 -07:00
Glenn Jocher
c78d49f190 check_file() update from yolov5 2020-06-15 12:25:48 -07:00
Glenn Jocher
002884ae5e multi_label burnin addition 2020-05-21 14:40:45 -07:00
Glenn Jocher
da40084b37 burnin update 2020-05-17 21:03:36 -07:00
Glenn Jocher
316d99c377 yolov5 regress updates to yolov3 2020-05-17 15:19:33 -07:00
Glenn Jocher
c94019f159 iglob bug fix 2020-05-17 14:31:14 -07:00
Glenn Jocher
37bd5490ef iglob file-search improvements 2020-05-16 22:25:21 -07:00
Glenn Jocher
0cf88f046d hyp evolution bug fix #1160 2020-05-12 09:53:13 -07:00
Glenn Jocher
d405959893 cleanup 2020-05-04 13:33:34 -07:00
Josh Veitch-Michaelis
fb1b5e09b2
faster and more informative training plots (#1114)
* faster and more informative training plots

* Update utils.py

Looks good. Needs pep8 linting, I'll do that in PyCharm later once PR is in.

* Update test.py

* Update train.py

f for the tb descriptor lets us plot several batches, i.e. to allow us to change L292 to 'if ni < 3' for 3 examples.

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-04-30 13:37:04 -07:00
Glenn Jocher
f1d73a29e5 Optimizer group report 2020-04-30 12:26:02 -07:00
Glenn Jocher
d62d68929c cleanup 2020-04-29 12:00:30 -07:00
Glenn Jocher
9f88f5cc21 cleanup 2020-04-29 11:34:59 -07:00
Glenn Jocher
37cbe89ef0 test/train jpg for png 2020-04-28 13:45:27 -07:00
Glenn Jocher
8521c3cff9 cleanup 2020-04-27 15:22:36 -07:00
Glenn Jocher
5a8efa5c1d auto --accumulate 2020-04-23 14:32:28 -07:00
Glenn Jocher
b3dfd89878 scheduler resume bug fix 2020-04-23 10:35:08 -07:00
Glenn Jocher
345d65d18f updated train default img_size 320-640 2020-04-22 14:32:05 -07:00
Glenn Jocher
7cbac5a3ea train.py iou_t to 0.20 2020-04-22 11:34:34 -07:00
Glenn Jocher
77b3829d56 check_git_status() to train.py 2020-04-22 11:02:09 -07:00
Glenn Jocher
a5bd0fe567 tensorboard comment=opt.name 2020-04-21 12:19:29 -07:00
Glenn Jocher
be3f322375 Tensorboard out of try, iou_t to 0.10 2020-04-20 09:57:15 -07:00
Glenn Jocher
c3edf8daf4 move image size report 2020-04-15 22:03:51 -07:00
Glenn Jocher
510eadcfa5 Apex and 'git pull' suggestions 2020-04-15 12:54:56 -07:00
Glenn Jocher
f5a2682a81 image sizes report 2020-04-14 12:02:08 -07:00
Glenn Jocher
763cdd5ae2 detailed image sizes report 2020-04-14 11:51:19 -07:00
Glenn Jocher
1038b0d269 multi-scale update 2020-04-12 18:22:54 -07:00
Timothy M. Shead
ada2958105
Fix argparse string escapes in train.py. (#1045) 2020-04-12 10:00:50 -07:00
Glenn Jocher
6e19245dc8 auto strip optimizer from best.pt after training 2020-04-09 19:53:29 -07:00
Glenn Jocher
d1601ae0f3 training updates 2020-04-08 21:34:34 -07:00
Glenn Jocher
4120ac3aa6 training updates 2020-04-08 21:01:58 -07:00
Glenn Jocher
97780cfdb4 parameterize grid size 2020-04-08 10:14:33 -07:00
Glenn Jocher
4fa3fd2df3 tensorboard notice 2020-04-06 16:33:23 -07:00
Glenn Jocher
05ae6e8499 tensorboard/focal loss reporting update 2020-04-06 15:45:18 -07:00
Glenn Jocher
c7f93bae40 update coco-tuned hyp['cls'] to current dataset 2020-04-06 10:58:07 -07:00
Glenn Jocher
e81a152a92 tensorboard notice and model verbose option 2020-04-05 13:35:58 -07:00
Glenn Jocher
41246aa042 tensorboard updates 2020-04-04 19:34:39 -07:00