-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
381 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,381 @@ | ||
[net] | ||
# Testing | ||
#batch=1 | ||
#subdivisions=1 | ||
# Training | ||
batch=64 | ||
subdivisions=4 | ||
width=416 | ||
height=416 | ||
channels=3 | ||
momentum=0.9 | ||
decay=0.0005 | ||
angle=0 | ||
saturation = 1.5 | ||
exposure = 1.5 | ||
hue=.1 | ||
|
||
learning_rate=0.0001 | ||
burn_in=100 | ||
max_batches = 4000 | ||
policy=steps | ||
steps=50000,60000 | ||
scales=.1,.1 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=32 | ||
size=3 | ||
stride=2 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=3 | ||
stride=2 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers=-1 | ||
groups=2 | ||
group_id=1 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=32 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=32 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1,-2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -6,-1 | ||
|
||
[maxpool] | ||
size=2 | ||
stride=2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers=-1 | ||
groups=2 | ||
group_id=1 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1,-2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -6,-1 | ||
|
||
[maxpool] | ||
size=2 | ||
stride=2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers=-1 | ||
groups=2 | ||
group_id=1 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1,-2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -6,-1 | ||
|
||
[maxpool] | ||
size=2 | ||
stride=2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=512 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
########### | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=512 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
filters=8 | ||
activation=linear | ||
|
||
[yolo] | ||
mask = 0 | ||
anchors = 100, 100 | ||
classes=3 | ||
num=1 | ||
jitter=.3 | ||
ignore_thresh = .7 | ||
truth_thresh = 1 | ||
random=1 | ||
|
||
[route] | ||
layers = -4 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[upsample] | ||
stride=2 | ||
|
||
[route] | ||
layers = -1, 24 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[convolutional] | ||
size=1 | ||
stride=1 | ||
pad=1 | ||
filters=8 | ||
activation=linear | ||
|
||
[yolo] | ||
mask = 0 | ||
anchors = 100, 100 | ||
classes=3 | ||
num=1 | ||
jitter=.3 | ||
ignore_thresh = .7 | ||
truth_thresh = 1 | ||
random=1 | ||
|
||
[route] | ||
layers = 18 | ||
groups=2 | ||
group_id=1 | ||
|
||
[maxpool] | ||
size=13 | ||
stride=13 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[upsample] | ||
stride=13 | ||
|
||
[route] | ||
layers = 18 | ||
groups=2 | ||
group_id=0 | ||
|
||
[route] | ||
layers = -1, -2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[shortcut] | ||
from=-2 | ||
|
||
[deconvolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=2 | ||
stride=2 | ||
pad=0 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1, 10 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=256 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[shortcut] | ||
from=-2 | ||
|
||
[deconvolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=2 | ||
stride=2 | ||
pad=0 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1, 2 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=128 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[shortcut] | ||
from=-2 | ||
|
||
[deconvolutional] | ||
batch_normalize=1 | ||
filters=32 | ||
size=2 | ||
stride=2 | ||
pad=0 | ||
activation=leaky | ||
|
||
[route] | ||
layers = -1, 0 | ||
|
||
[convolutional] | ||
batch_normalize=1 | ||
filters=64 | ||
size=3 | ||
stride=1 | ||
pad=1 | ||
activation=leaky | ||
|
||
[shortcut] | ||
from=-2 | ||
|
||
[deconvolutional] | ||
batch_normalize=1 | ||
filters=3 | ||
size=2 | ||
stride=2 | ||
pad=0 | ||
activation=linear | ||
|
||
[seg] | ||
classes=3 |