Glenn Jocher
|
628028c617
|
bias init
|
2020-04-15 11:50:54 -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
|
46726dad13
|
torch.tensor(ng, device=device)
|
2020-04-12 13:02:00 -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
|
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
|
b6959a2f54
|
ONNX export self.training=False
|
2020-04-08 10:25:52 -07:00 |
|
Glenn Jocher
|
933e05cb44
|
augment update
|
2020-04-07 17:35:35 -07:00 |
|
Glenn Jocher
|
b3e1d74478
|
remove imwrite from augment
|
2020-04-07 14:23:31 -07:00 |
|
Glenn Jocher
|
d79c3bd076
|
parameterize augment scales
|
2020-04-07 14:19:43 -07:00 |
|
Glenn Jocher
|
1a511d2906
|
updated --augment sizes and results
|
2020-04-07 12:27:49 -07:00 |
|
Glenn Jocher
|
4fc0012829
|
initial batchnorm to 0.03 momentum
|
2020-04-05 18:03:49 -07:00 |
|
Glenn Jocher
|
2baf4e3f93
|
imagenet normalization on layer 0 batchnorm2d()
|
2020-04-05 17:33:06 -07:00 |
|
Glenn Jocher
|
c6d4e80335
|
move inference augmentation to model.forward()
|
2020-04-05 17:14:26 -07:00 |
|
Glenn Jocher
|
bb59ffe68f
|
model forward() zip() removal
|
2020-04-05 15:22:32 -07:00 |
|
Glenn Jocher
|
a657345b45
|
add FeatureConcat() module
|
2020-04-05 14:47:41 -07:00 |
|
Glenn Jocher
|
d04738a27c
|
forward updated if-else
|
2020-04-05 13:49:13 -07:00 |
|
Glenn Jocher
|
e81a152a92
|
tensorboard notice and model verbose option
|
2020-04-05 13:35:58 -07:00 |
|
Glenn Jocher
|
00c1fdd805
|
add MixConv2d() layer
|
2020-04-03 20:03:44 -07:00 |
|
Glenn Jocher
|
eb9fb245aa
|
add support for standalone BatchNorm2d()
|
2020-04-03 14:21:47 -07:00 |
|
Glenn Jocher
|
682c2b27e7
|
smart bias bug fix
|
2020-04-03 12:42:09 -07:00 |
|
Glenn Jocher
|
41a002e798
|
grid.float()
|
2020-04-03 12:38:08 -07:00 |
|
Glenn Jocher
|
93055a9d58
|
create_grids() to YOLOLayer method
|
2020-04-02 20:23:55 -07:00 |
|
Glenn Jocher
|
91f563c2a2
|
create_grids() to YOLOLayer method
|
2020-04-02 19:10:51 -07:00 |
|
Glenn Jocher
|
aa4591d7e9
|
batchnorm momentum to 0.03
|
2020-04-02 14:10:45 -07:00 |
|
Glenn Jocher
|
27c7334e81
|
new layers.py file
|
2020-04-02 12:22:15 -07:00 |
|
Glenn Jocher
|
98271eb6ed
|
remove deprecated models
|
2020-03-31 14:27:10 -07:00 |
|
Glenn Jocher
|
ac2aa56e0a
|
feature fusion update
|
2020-03-30 17:53:17 -07:00 |
|
Glenn Jocher
|
89b6377723
|
Fuse by default when test.py called directly (faster)
|
2020-03-19 18:11:08 -07:00 |
|
Glenn Jocher
|
fff45c39a8
|
cleanup/speedup
|
2020-03-19 16:41:42 -07:00 |
|
Glenn Jocher
|
1b68fe7fde
|
cleanup
|
2020-03-19 16:23:44 -07:00 |
|
Glenn Jocher
|
448c4a6e1f
|
Remove deprecated --arc architecture options, implement --arc default for all cases
|
2020-03-16 20:46:25 -07:00 |
|
Glenn Jocher
|
5ebbb2db28
|
ASFF implementation
|
2020-03-14 17:04:38 -07:00 |
|
Glenn Jocher
|
ea4c26b32d
|
BatchNorm2d() to EfficientDet defaults: decay=0.997 eps=1e-4
|
2020-03-14 16:54:04 -07:00 |
|
Glenn Jocher
|
9ce4ec48a7
|
model.info() method implemented
|
2020-03-14 16:46:54 -07:00 |
|
Glenn Jocher
|
d55dbc1f29
|
updates
|
2020-03-09 20:08:19 -07:00 |
|
Glenn Jocher
|
17a06dcf83
|
updates
|
2020-03-09 18:55:17 -07:00 |
|
Glenn Jocher
|
d8370d13ea
|
updates
|
2020-03-09 18:49:35 -07:00 |
|
Glenn Jocher
|
f7f435446b
|
updates
|
2020-03-09 18:24:20 -07:00 |
|
Glenn Jocher
|
25ad727a3d
|
updates
|
2020-03-09 18:22:42 -07:00 |
|
Glenn Jocher
|
207cf14df4
|
updates
|
2020-03-09 18:03:34 -07:00 |
|
Glenn Jocher
|
204594f299
|
updates
|
2020-03-09 16:44:26 -07:00 |
|
Glenn Jocher
|
6130b70fe7
|
updates
|
2020-03-09 16:00:05 -07:00 |
|
Glenn Jocher
|
67e7ac221f
|
updates
|
2020-03-09 14:20:38 -07:00 |
|
Glenn Jocher
|
cd76a1a982
|
updates
|
2020-03-09 10:46:59 -07:00 |