985 Commits

Author SHA1 Message Date
Glenn Jocher
e80cc2b80e
Update datasets.py 2020-07-20 10:34:06 -07:00
Glenn Jocher
f61fa7de2b
Update datasets.py 2020-07-20 10:33:49 -07:00
tjiagoM
fa78fc4e34
partial support for dropout layer (#1366) 2020-07-02 14:35:20 -07:00
Glenn Jocher
183e3833d2 update datasets.py 2020-06-20 09:58:48 -07:00
Glenn Jocher
c78d49f190 check_file() update from yolov5 2020-06-15 12:25:48 -07:00
Glenn Jocher
cf7a4d31d3 bug fix in local to global path replacement 2020-05-28 20:50:02 -07:00
Glenn Jocher
e99ff3aad0 local path robustness 2020-05-28 14:01:38 -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
16ea613628 caching introspection update 2020-05-22 16:06:21 -07:00
Glenn Jocher
4879fd22e9 caching introspection update 2020-05-22 16:03:08 -07:00
Glenn Jocher
2cc2b2cf0d label *.npy saving for faster caching 2020-05-20 21:39:18 -07:00
Glenn Jocher
3ddaf3b63c label *.npy saving for faster caching 2020-05-20 21:13:41 -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
c6ea2b58ea auto-accumulate update 2020-04-28 15:06:33 -07:00
Josh Veitch-Michaelis
992d8af242
faster hsv augmentation (#1110)
As per https://github.com/ultralytics/yolov3/issues/1096
2020-04-28 12:59:44 -07:00
Glenn Jocher
15f1343dfc uncached label removal 2020-04-28 11:07:26 -07:00
Glenn Jocher
2518868508 MemoryEfficientMish() 2020-04-27 13:51:21 -07:00
Glenn Jocher
3aa347a321 add HardSwish() 2020-04-27 13:08:24 -07:00
Glenn Jocher
4a4bfb20de FLOPS verbose=False 2020-04-26 16:31:57 -07:00
Glenn Jocher
a0a3bab9e6 add Mish() support 2020-04-26 16:31:21 -07:00
Glenn Jocher
754a1b5bf8 reduce merge limit to 3000 2020-04-25 21:15:30 -07:00
Glenn Jocher
82a12e2c8e docker train update 2020-04-22 11:38:48 -07:00
Glenn Jocher
77b3829d56 check_git_status() to train.py 2020-04-22 11:02:09 -07:00
Glenn Jocher
03c6a2d6fa cleanup 2020-04-21 12:31:37 -07:00
Glenn Jocher
accce6b565 git status check bug fix 2020-04-18 18:06:11 -07:00
Glenn Jocher
693c06b26c bug fix issues/1067 2020-04-18 12:07:44 -07:00
Glenn Jocher
bf1061c146 cleanup 2020-04-16 16:12:23 -07:00
Glenn Jocher
510eadcfa5 Apex and 'git pull' suggestions 2020-04-15 12:54:56 -07:00
Glenn Jocher
a49ea80218 update initialize_weights() 2020-04-14 15:58:32 -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
835b0da68a new modules and init weights 2020-04-14 01:20:57 -07:00