Glenn Jocher
|
628028c617
|
bias init
|
2020-04-15 11:50:54 -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
|
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
|
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
|
0dd5f8eee8
|
code cleanup
|
2020-04-13 18:25:59 -07:00 |
|
Glenn Jocher
|
ca3a9fcb0b
|
return get_yolo_layers()
|
2020-04-13 17:56:12 -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
|
1038b0d269
|
multi-scale update
|
2020-04-12 18:22:54 -07:00 |
|
Glenn Jocher
|
46726dad13
|
torch.tensor(ng, device=device)
|
2020-04-12 13:02:00 -07:00 |
|
Glenn Jocher
|
efc754a794
|
add generations arg to kmeans()
|
2020-04-12 12:49:23 -07:00 |
|
Glenn Jocher
|
f65a50d13d
|
Merge remote-tracking branch 'origin/master'
|
2020-04-12 10:20:28 -07:00 |
|
Glenn Jocher
|
eda31c8bd0
|
print speeds for save_json
|
2020-04-12 10:20:21 -07:00 |
Timothy M. Shead
|
ada2958105
|
Fix argparse string escapes in train.py. (#1045)
|
2020-04-12 10:00:50 -07:00 |
|
Glenn Jocher
|
ed1d4f5ae7
|
k for kernel_size
|
2020-04-11 12:37:03 -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
|
7be71b02e2
|
get_yolo_layers()
|
2020-04-11 10:56:20 -07:00 |
|
Glenn Jocher
|
dcc2e99fb2
|
get_yolo_layers()
|
2020-04-11 10:55:49 -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 |
|
Glenn Jocher
|
9bc3a551d9
|
histogram equalization added to augmentation
|
2020-04-10 17:24:49 -07:00 |
|
Glenn Jocher
|
398f8eadec
|
add thr=0.10 to kmean_anchors()
|
2020-04-10 16:34:32 -07:00 |
|
Glenn Jocher
|
aa8b1098dd
|
adapt mosaic to img channel count
|
2020-04-10 16:28:59 -07:00 |
|
Glenn Jocher
|
6736d7d125
|
swap cv2.INTER_AREA for cv2.INTER_LINEAR
|
2020-04-10 12:47:07 -07:00 |
|
Glenn Jocher
|
b98ce11d3a
|
add MixConv2d() layer
|
2020-04-09 20:20:23 -07:00 |
|
Glenn Jocher
|
6e19245dc8
|
auto strip optimizer from best.pt after training
|
2020-04-09 19:53:29 -07:00 |
|
Glenn Jocher
|
bc74822540
|
notebook update
|
2020-04-09 14:33:24 -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
|
c7f5b6cc21
|
speed update
|
2020-04-08 20:53:17 -07:00 |
|
Glenn Jocher
|
d9af081b7f
|
Merge remote-tracking branch 'origin/master'
|
2020-04-08 20:43:57 -07:00 |
|
Glenn Jocher
|
8f71b7865b
|
run once to remove initial timing effects
|
2020-04-08 20:43:51 -07:00 |
|
Glenn Jocher
|
4f6843e249
|
Update README.md
|
2020-04-08 12:28:10 -07:00 |
|
Glenn Jocher
|
13fa2798a6
|
Update README.md
|
2020-04-08 12:27:33 -07:00 |
|
Glenn Jocher
|
b6959a2f54
|
ONNX export self.training=False
|
2020-04-08 10:25:52 -07:00 |
|
Glenn Jocher
|
97780cfdb4
|
parameterize grid size
|
2020-04-08 10:14:33 -07:00 |
|
Glenn Jocher
|
9f41b7601a
|
inference speed and mAP updates
|
2020-04-08 10:02:20 -07:00 |
|
Glenn Jocher
|
f54d28ba63
|
improve assert no labels found
|
2020-04-08 09:57:59 -07:00 |
|
Glenn Jocher
|
933e05cb44
|
augment update
|
2020-04-07 17:35:35 -07:00 |
|
Glenn Jocher
|
6c5ecaf805
|
remove label loading during training
|
2020-04-07 16:57:22 -07:00 |
|
Glenn Jocher
|
b3e1d74478
|
remove imwrite from augment
|
2020-04-07 14:23:31 -07:00 |