adding coco models

This commit is contained in:
Joseph Redmon 2016-09-08 00:04:39 -07:00
parent a19604bf5f
commit c62b4f35aa
2 changed files with 161 additions and 21 deletions

125
cfg/tiny-coco.cfg Normal file
View File

@ -0,0 +1,125 @@
[net]
batch=64
subdivisions=2
height=448
width=448
channels=3
momentum=0.9
decay=0.0005
hue = .1
saturation=.75
exposure=.75
learning_rate=0.0005
policy=steps
steps=200,400,600,800,100000,150000
scales=2.5,2,2,2,.1,.1
max_batches = 200000
[convolutional]
batch_normalize=1
filters=16
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=32
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=64
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=128
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=256
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
size=3
stride=1
pad=1
filters=1024
activation=leaky
[convolutional]
batch_normalize=1
size=3
stride=1
pad=1
filters=256
activation=leaky
[connected]
output= 4655
activation=linear
[detection]
classes=80
coords=4
rescore=1
side=7
num=3
softmax=0
sqrt=1
jitter=.2
object_scale=1
noobject_scale=.5
class_scale=1
coord_scale=5

View File

@ -7,21 +7,18 @@ channels=3
momentum=0.9 momentum=0.9
decay=0.0005 decay=0.0005
learning_rate=0.0001 hue = .1
policy=steps saturation=.75
steps=100,200,300,100000,150000 exposure=.75
scales=2.5,2,2,.1,.1
max_batches = 300000
[crop] learning_rate=0.0005
crop_width=448 policy=steps
crop_height=448 steps=200,400,600,800,100000,150000
flip=0 scales=2.5,2,2,2,.1,.1
angle=0 max_batches = 200000
saturation = 1.5
exposure = 1.5
[convolutional] [convolutional]
batch_normalize=1
filters=64 filters=64
size=7 size=7
stride=2 stride=2
@ -33,6 +30,7 @@ size=2
stride=2 stride=2
[convolutional] [convolutional]
batch_normalize=1
filters=192 filters=192
size=3 size=3
stride=1 stride=1
@ -44,6 +42,7 @@ size=2
stride=2 stride=2
[convolutional] [convolutional]
batch_normalize=1
filters=128 filters=128
size=1 size=1
stride=1 stride=1
@ -51,6 +50,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=3 size=3
stride=1 stride=1
@ -58,6 +58,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=1 size=1
stride=1 stride=1
@ -65,6 +66,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=3 size=3
stride=1 stride=1
@ -76,6 +78,7 @@ size=2
stride=2 stride=2
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=1 size=1
stride=1 stride=1
@ -83,6 +86,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=3 size=3
stride=1 stride=1
@ -90,6 +94,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=1 size=1
stride=1 stride=1
@ -97,6 +102,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=3 size=3
stride=1 stride=1
@ -104,6 +110,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=1 size=1
stride=1 stride=1
@ -111,6 +118,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=3 size=3
stride=1 stride=1
@ -118,6 +126,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=256 filters=256
size=1 size=1
stride=1 stride=1
@ -125,6 +134,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=3 size=3
stride=1 stride=1
@ -132,6 +142,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=1 size=1
stride=1 stride=1
@ -139,6 +150,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=1024 filters=1024
size=3 size=3
stride=1 stride=1
@ -150,6 +162,7 @@ size=2
stride=2 stride=2
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=1 size=1
stride=1 stride=1
@ -157,6 +170,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=1024 filters=1024
size=3 size=3
stride=1 stride=1
@ -164,6 +178,7 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=512 filters=512
size=1 size=1
stride=1 stride=1
@ -171,17 +186,17 @@ pad=1
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
filters=1024 filters=1024
size=3 size=3
stride=1 stride=1
pad=1 pad=1
activation=leaky activation=leaky
#######
#################################
[convolutional] [convolutional]
batch_normalize=1
size=3 size=3
stride=1 stride=1
pad=1 pad=1
@ -189,6 +204,7 @@ filters=1024
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
size=3 size=3
stride=2 stride=2
pad=1 pad=1
@ -196,6 +212,7 @@ filters=1024
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
size=3 size=3
stride=1 stride=1
pad=1 pad=1
@ -203,6 +220,7 @@ filters=1024
activation=leaky activation=leaky
[convolutional] [convolutional]
batch_normalize=1
size=3 size=3
stride=1 stride=1
pad=1 pad=1
@ -213,14 +231,11 @@ activation=leaky
size=3 size=3
stride=1 stride=1
pad=1 pad=1
filters=192 filters=256
activation=leaky activation=leaky
[dropout]
probability=.5
[connected] [connected]
output= 4410 output= 4655
activation=linear activation=linear
[detection] [detection]
@ -228,7 +243,7 @@ classes=80
coords=4 coords=4
rescore=1 rescore=1
side=7 side=7
num=2 num=3
softmax=0 softmax=0
sqrt=1 sqrt=1
jitter=.2 jitter=.2