Glenn Jocher
|
20891926c9
|
add stride order reversal for c53*.cfg
|
2020-05-15 14:40:14 -07:00 |
|
Glenn Jocher
|
6fe67595cb
|
add stride order reversal for c53*.cfg
|
2020-05-15 11:32:23 -07:00 |
IlyaOvodov
|
b2fcfc573e
|
convert(...) changed to save converted file alongside the original file (#1167)
|
2020-05-13 09:08:55 -07:00 |
|
Glenn Jocher
|
9cc4951d4f
|
auto reverse-strides for yolov4/panet
|
2020-04-28 15:24:14 -07:00 |
|
Glenn Jocher
|
18d4ebfd12
|
add Mish() support
|
2020-04-26 16:25:46 -07:00 |
|
Glenn Jocher
|
efbeb283c4
|
ONNX grid float
|
2020-04-26 14:01:20 -07:00 |
|
Glenn Jocher
|
6aed6c5fd0
|
attempt_download() update for '' weights
|
2020-04-21 09:13:38 -07:00 |
|
Glenn Jocher
|
b8c3644a18
|
ONNX export update
|
2020-04-15 12:12:59 -07:00 |
|
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 |