Commit Graph

539 Commits

Author SHA1 Message Date
tomasz ebd423139c Update 2020-06-30 21:51:10 +02:00
Glenn Jocher c78d49f190 check_file() update from yolov5 2020-06-15 12:25:48 -07:00
Glenn Jocher d6d6fb5e5b print('Optimizer stripped from %s' % f) 2020-05-24 20:30:30 -07:00
Glenn Jocher 23f85a68b8 tight_layout=True 2020-05-24 10:51:35 -07:00
Glenn Jocher eacded6a2c add stride order reversal for c53*.cfg 2020-05-17 22:45:48 -07:00
Glenn Jocher bc9da228e0 add stride order reversal for c53*.cfg 2020-05-17 22:11:02 -07:00
Glenn Jocher 0c7d7427e4 [conf > conf_thres] update 2020-05-17 20:59:19 -07:00
Glenn Jocher 316d99c377 yolov5 regress updates to yolov3 2020-05-17 15:19:33 -07:00
Glenn Jocher c8f4ee6c46 yolov5 regress updates to yolov3 - build_targets() 2020-05-17 15:10:31 -07:00
Glenn Jocher 110ead20e6 yolov5 regress updates to yolov3 2020-05-17 15:00:07 -07:00
Glenn Jocher bbd82bb94d updates 2020-05-17 14:30:12 -07:00
Glenn Jocher 9f04e175f6 nms torch.mm() update 2020-05-10 11:26:37 -07:00
Glenn Jocher ae2bc020eb git status check - linux and darwin 2020-05-09 22:35:44 -07:00
Glenn Jocher 965155ee60 CUBLAS bug fix #1139 2020-05-06 10:26:28 -07:00
Glenn Jocher ee7cba65a5 kmeans() cleanup 2020-05-02 09:20:19 -07:00
Glenn Jocher b0629d622c bug fix on #1114 2020-04-30 15:03:32 -07:00
Glenn Jocher 0ffbf5534e cleanup for #1114 2020-04-30 14:53:57 -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 754a1b5bf8 reduce merge limit to 3000 2020-04-25 21:15:30 -07:00
Glenn Jocher 77b3829d56 check_git_status() to train.py 2020-04-22 11:02:09 -07:00
Glenn Jocher accce6b565 git status check bug fix 2020-04-18 18:06:11 -07:00
Glenn Jocher 510eadcfa5 Apex and 'git pull' suggestions 2020-04-15 12:54:56 -07:00
Glenn Jocher ac4c90c817 cleanup 2020-04-14 13:08:00 -07:00
Glenn Jocher 763cdd5ae2 detailed image sizes report 2020-04-14 11:51:19 -07:00
Glenn Jocher 029e137bc2 bug fix 2020-04-14 04:34:40 -07:00
Glenn Jocher 1681249588 cleanup 2020-04-14 04:15:53 -07:00
Glenn Jocher 198a5a591d code cleanup 2020-04-14 04:15:05 -07:00
Glenn Jocher 25725c8569 bug fix 2020-04-14 03:13:30 -07:00
Glenn Jocher 76fb8d48d4 ng dependence removed from build_targets() 2020-04-13 21:25:03 -07:00
Glenn Jocher efc754a794 add generations arg to kmeans() 2020-04-12 12:49:23 -07:00
Glenn Jocher b574f765ce add warning to plot_results() 2020-04-11 11:04:10 -07:00
Glenn Jocher 58edfc4a84 kaiming weight init 2020-04-11 10:45:33 -07:00
Glenn Jocher 398f8eadec add thr=0.10 to kmean_anchors() 2020-04-10 16:34:32 -07:00
Glenn Jocher 4120ac3aa6 training updates 2020-04-08 21:01:58 -07:00
Glenn Jocher a19b1a3b94 line thickness 2020-04-05 11:10:05 -07:00
Glenn Jocher 6203340888 detect.py multi_label default False 2020-04-05 11:05:49 -07:00
Glenn Jocher 207c6fcff9 merge NMS full matrix 2020-04-02 18:53:40 -07:00
Glenn Jocher 300e9a7ad6 merge NMS full matrix 2020-03-31 21:31:09 -07:00
Glenn Jocher 02802e67f2 merge NMS full matrix 2020-03-31 18:18:08 -07:00
Glenn Jocher f4eecef700 merge NMS speed/memory improvements 2020-03-31 15:37:23 -07:00
Glenn Jocher eb151a881e NMS and test batch_size updates 2020-03-29 20:41:32 -07:00
Glenn Jocher f9d34587da Merge NMS update 2020-03-27 13:11:24 -07:00
GoogleWiki 582de735ad
utils.clip_coords doesn't work as expected. (#961)
* utils.clip_coords doesn't work as expected.

Box coords may be negative or exceed borders.

* Update utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-27 13:09:10 -07:00
Glenn Jocher 4a63b24b09 Merge NMS update 2020-03-26 19:50:29 -07:00
Glenn Jocher 01ee0c5e95 Merge NMS update 2020-03-26 18:41:04 -07:00
Glenn Jocher dad59220f1 speed and comments update 2020-03-26 18:34:20 -07:00
Glenn Jocher a322fc5d4b Merge NMS update 2020-03-26 12:48:00 -07:00
Glenn Jocher 171b4129b5 Merge NMS update 2020-03-26 12:33:12 -07:00
Glenn Jocher eac07f9da3 Merge NMS update 2020-03-26 12:20:01 -07:00
Glenn Jocher 23b34f4db8 merge_batch NMS method 2020-03-25 23:29:33 -07:00