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 |
|
Glenn Jocher
|
d79c3bd076
|
parameterize augment scales
|
2020-04-07 14:19:43 -07:00 |
|
Glenn Jocher
|
b9b14bef59
|
scale_img() bug fix
|
2020-04-07 13:35:47 -07:00 |
|
Glenn Jocher
|
067ee264c0
|
scale_img() bug fix
|
2020-04-07 12:59:52 -07:00 |
|
Glenn Jocher
|
68f58f4dec
|
scale_img() update
|
2020-04-07 12:51:52 -07:00 |
|
Glenn Jocher
|
1a3c77df95
|
Merge remote-tracking branch 'origin/master'
|
2020-04-07 12:27:56 -07:00 |
|
Glenn Jocher
|
1a511d2906
|
updated --augment sizes and results
|
2020-04-07 12:27:49 -07:00 |
Wang Xinyu
|
b20e3c4c40
|
Update README, add link to yolov3-spp-tensorrt (#1017)
|
2020-04-06 22:56:39 -07:00 |
|
Glenn Jocher
|
4fa3fd2df3
|
tensorboard notice
|
2020-04-06 16:33:23 -07:00 |
|
Glenn Jocher
|
05ae6e8499
|
tensorboard/focal loss reporting update
|
2020-04-06 15:45:18 -07:00 |
|
Glenn Jocher
|
c7f93bae40
|
update coco-tuned hyp['cls'] to current dataset
|
2020-04-06 10:58:07 -07:00 |
|
Glenn Jocher
|
4fc0012829
|
initial batchnorm to 0.03 momentum
|
2020-04-05 18:03:49 -07:00 |
|
Glenn Jocher
|
26fc4fb018
|
dataloader default color to imagenet mean 114
|
2020-04-05 18:02:41 -07:00 |
|
Glenn Jocher
|
2baf4e3f93
|
imagenet normalization on layer 0 batchnorm2d()
|
2020-04-05 17:33:06 -07:00 |
|
Glenn Jocher
|
b70cfa9a29
|
mAP updates for rect inference 64 commit
|
2020-04-05 17:18:24 -07:00 |
|
Glenn Jocher
|
c6d4e80335
|
move inference augmentation to model.forward()
|
2020-04-05 17:14:26 -07:00 |
|
Glenn Jocher
|
4da5c6c114
|
rect padding to 64, mAP increase 42.7 to 42.9
|
2020-04-05 16:06:27 -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
|
968b2ec004
|
.fuse() after .eval()
|
2020-04-05 14:05:12 -07:00 |
|
Glenn Jocher
|
d04738a27c
|
forward updated if-else
|
2020-04-05 13:49:13 -07:00 |