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
Glenn Jocher
76fb8d48d4
ng dependence removed from build_targets()
2020-04-13 21:25:03 -07:00
Glenn Jocher
b8574add37
new find_modules() fcn
2020-04-13 17:48:30 -07:00
Glenn Jocher
77e6bdd3c1
FLOPs at 480x640, BN init
2020-04-12 18:44:18 -07:00
Glenn Jocher
efc754a794
add generations arg to kmeans()
2020-04-12 12:49:23 -07:00
Glenn Jocher
a34219a54b
pading from (k-1) // 2 to k // 2
2020-04-11 12:18:54 -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
2cf23c4aee
add MixConv2d() layer
2020-04-10 18:58:34 -07:00
Glenn Jocher
4bbfda5cde
hist equalization
2020-04-10 17:29:57 -07:00