new cfg files for classification

This commit is contained in:
Joseph Redmon 2015-06-11 15:51:17 -07:00
parent 11c72b1132
commit 0df9d25c46
2 changed files with 334 additions and 0 deletions

182
cfg/strided.cfg Normal file
View File

@ -0,0 +1,182 @@
[net]
batch=128
subdivisions=2
height=256
width=256
channels=3
learning_rate=0.01
momentum=0.9
decay=0.0005
seen=0
[crop]
crop_height=224
crop_width=224
flip=1
angle=15
saturation=1.5
exposure=1.5
[convolutional]
filters=64
size=7
stride=2
pad=1
activation=ramp
[convolutional]
filters=192
size=3
stride=2
pad=1
activation=ramp
[convolutional]
filters=128
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=3
stride=2
pad=1
activation=ramp
[convolutional]
filters=128
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=128
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=512
size=3
stride=2
pad=1
activation=ramp
[convolutional]
filters=256
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=ramp
[convolutional]
filters=256
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=1024
size=3
stride=2
pad=1
activation=ramp
[convolutional]
filters=512
size=1
stride=1
pad=1
activation=ramp
[convolutional]
filters=1024
size=3
stride=1
pad=1
activation=ramp
[maxpool]
size=3
stride=2
[dropout]
probability=0.5
[connected]
output=4096
activation=ramp
[dropout]
probability=0.5
[connected]
output=1000
activation=ramp
[softmax]
[cost]
type=sse

152
cfg/vgg-16.cfg Normal file
View File

@ -0,0 +1,152 @@
[net]
batch=128
subdivisions=8
height=256
width=256
channels=3
learning_rate=0.00001
momentum=0.9
decay=0.0005
[crop]
crop_height=224
crop_width=224
flip=1
exposure=2
saturation=2
angle=5
[convolutional]
filters=64
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=64
size=3
stride=1
pad=1
activation=relu
[maxpool]
size=2
stride=2
[convolutional]
filters=128
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=128
size=3
stride=1
pad=1
activation=relu
[maxpool]
size=2
stride=2
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=256
size=3
stride=1
pad=1
activation=relu
[maxpool]
size=2
stride=2
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[maxpool]
size=2
stride=2
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[convolutional]
filters=512
size=3
stride=1
pad=1
activation=relu
[maxpool]
size=2
stride=2
[connected]
output=4096
activation=relu
[dropout]
probability=.5
[connected]
output=4096
activation=relu
[dropout]
probability=.5
[connected]
output=1000
activation=linear
[softmax]
[cost]
type=sse