Skip to content

Commit

Permalink
SELU activation and yolo openimages
Browse files Browse the repository at this point in the history
  • Loading branch information
pjreddie committed Sep 5, 2018
1 parent 98e2a76 commit 680d3bd
Show file tree
Hide file tree
Showing 10 changed files with 1,445 additions and 17 deletions.
8 changes: 8 additions & 0 deletions cfg/openimages.data
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
classes= 601
train = /home/pjreddie/data/openimsv4/openimages.train.list
#valid = coco_testdev
valid = data/coco_val_5k.list
names = data/openimages.names
backup = /home/pjreddie/backup/
eval=coco

19 changes: 12 additions & 7 deletions cfg/resnext101-32x4d.cfg
Original file line number Diff line number Diff line change
@@ -1,21 +1,25 @@
[net]
batch=128
subdivisions=8
batch=1
subdivisions=1
# Training
# batch=128
# subdivisions=8

# Testing
batch=1
subdivisions=1

height=256
width=256
channels=3
min_crop=128
max_crop=448
channels=3
momentum=0.9
decay=0.0005

burn_in=1000
learning_rate=0.1
policy=poly
power=4
max_batches=800000
momentum=0.9
decay=0.0005

angle=7
hue=.1
Expand All @@ -24,6 +28,7 @@ exposure=.75
aspect=.75



[convolutional]
batch_normalize=1
filters=64
Expand Down
18 changes: 11 additions & 7 deletions cfg/resnext152-32x4d.cfg
Original file line number Diff line number Diff line change
@@ -1,21 +1,25 @@
[net]
batch=128
subdivisions=16
# batch=1
# subdivisions=1
# Training
# batch=128
# subdivisions=16

# Testing
batch=1
subdivisions=1

height=256
width=256
channels=3
min_crop=128
max_crop=448
channels=3
momentum=0.9
decay=0.0005

burn_in=1000
learning_rate=0.1
policy=poly
power=4
max_batches=800000
momentum=0.9
decay=0.0005

angle=7
hue=.1
Expand Down
10 changes: 8 additions & 2 deletions cfg/vgg-16.cfg
Original file line number Diff line number Diff line change
@@ -1,6 +1,12 @@
[net]
batch=128
subdivisions=4
# Training
# batch=128
# subdivisions=4

# Testing
batch=1
subdivisions=1

height=256
width=256
channels=3
Expand Down
Loading

0 comments on commit 680d3bd

Please sign in to comment.