inet label script

This commit is contained in:
Joseph Redmon 2015-06-12 22:25:18 -07:00
parent 0df9d25c46
commit d572dc7e66
5 changed files with 1019 additions and 3 deletions

1
.gitignore vendored
View File

@ -3,7 +3,6 @@
*.csv
*.out
*.png
*.sh
mnist/
data/
caffe/

View File

@ -1,5 +1,5 @@
GPU=0
OPENCV=0
GPU=1
OPENCV=1
DEBUG=0
ARCH= -arch=sm_52

View File

@ -6,6 +6,7 @@ width=256
channels=3
learning_rate=0.00001
momentum=0.9
seen=0
decay=0.0005
[crop]
@ -146,6 +147,7 @@ output=1000
activation=linear
[softmax]
groups=1
[cost]
type=sse

1000
data/inet.labels.list Normal file

File diff suppressed because it is too large Load Diff

15
scripts/label.sh Normal file
View File

@ -0,0 +1,15 @@
#!/bin/bash
mkdir -p labelled
wd=`pwd`
for f in val/*.xml;
do
label=`grep -m1 "<name>" $f | grep -oP '<name>\K[^<]*'`
im=`echo $f | sed 's/val/imgs/; s/xml/JPEG/'`
out=`echo $im | sed 's/JPEG/'${label}'.JPEG/; s/imgs/labelled/'`
ln -s ${wd}/$im ${wd}/$out
done
find ${wd}/labelled -name \*.JPEG > inet.val.list