Add data dir

.gitignore

@@ -61,7 +61,7 @@ Makefile.config
 # Data and models are either
 # 1. reference, and not casually committed
 # 2. custom, and live on their own unless they're deliberated contributed
-data/*
+#data/*
 #models/*
 #*.caffemodel
 *.caffemodel.h5

README.md

@@ -6,7 +6,7 @@ A caffe implementation of MobileNet-YOLO (YOLOv2 base) detection network, with p
 
 Network|mAP|Download|Download
 :---:|:---:|:---:|:---:
-MobileNet-YOLO-Lite|0.66|[train](models/MobileNet/mobilenet_iter_73000.caffemodel)|[Deploy](models/yolov2/mobilenet_yolo_deploy_iter_95000.caffemodel)
+MobileNet-YOLO-Lite|0.66|[train](models/MobileNet/mobilenet_iter_73000.caffemodel)|[deploy](models/yolov2/mobilenet_yolo_deploy_iter_95000.caffemodel)
 
 ## Performance
 

data/VOC0712/coco_voc_map.txt

@@ -0,0 +1,21 @@
+0,0,background
+5,1,aeroplane
+2,2,bicycle
+15,3,bird
+9,4,boat
+40,5,bottle
+6,6,bus
+3,7,car
+16,8,cat
+57,9,chair
+20,10,cow
+61,11,diningtable
+17,12,dog
+18,13,horse
+4,14,motorbike
+1,15,person
+59,16,pottedplant
+19,17,sheep
+58,18,sofa
+7,19,train
+63,20,tvmonitor

data/VOC0712/create_data.sh

@@ -0,0 +1,28 @@
+cur_dir=$(cd $( dirname ${BASH_SOURCE[0]} ) && pwd )
+root_dir=$cur_dir/../..
+
+cd $root_dir
+
+redo=1
+data_root_dir="data/VOCdevkit"
+dataset_name="VOC0712"
+mapfile="$root_dir/data/$dataset_name/labelmap_voc.prototxt"
+anno_type="detection"
+db="lmdb"
+min_dim=0
+max_dim=0
+width=0
+height=0
+
+extra_cmd="--encode-type=jpg --encoded"
+if [ $redo ]
+then
+  extra_cmd="$extra_cmd --redo"
+fi
+for subset in test trainval
+do
+  python $root_dir/scripts/create_annoset.py --anno-type=$anno_type \
+  --label-map-file=$mapfile --min-dim=$min_dim --max-dim=$max_dim --resize-width=$width \
+  --resize-height=$height --check-label $extra_cmd $data_root_dir $root_dir/data/$dataset_name/$subset.txt \
+  $data_root_dir/$dataset_name/$db/$dataset_name"_"$subset"_"$db examples/$dataset_name
+done

data/VOC0712/create_data_fisheye.sh

@@ -0,0 +1,28 @@
+cur_dir=$(cd $( dirname ${BASH_SOURCE[0]} ) && pwd )
+root_dir=$cur_dir/../..
+
+cd $root_dir
+
+redo=1
+data_root_dir="data/VOCdevkit"
+dataset_name="fisheye"
+mapfile="$root_dir/data/$dataset_name/labelmap_fisheye.prototxt"
+anno_type="detection"
+db="lmdb"
+min_dim=0
+max_dim=0
+width=0
+height=0
+
+extra_cmd="--encode-type=jpg --encoded"
+if [ $redo ]
+then
+  extra_cmd="$extra_cmd --redo"
+fi
+for subset in test trainval
+do
+  python $root_dir/scripts/create_annoset.py --anno-type=$anno_type \
+  --label-map-file=$mapfile --min-dim=$min_dim --max-dim=$max_dim --resize-width=$width \
+  --resize-height=$height --check-label $extra_cmd $data_root_dir $root_dir/data/$dataset_name/$subset.txt \
+  $data_root_dir/$dataset_name/$db/$dataset_name"_"$subset"_"$db examples/$dataset_name
+done

data/VOC0712/create_list.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+
+root_dir="../../data/VOCdevkit"
+sub_dir=ImageSets/Main
+bash_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+for dataset in trainval test
+do
+  dst_file=$bash_dir/$dataset.txt
+  if [ -f $dst_file ]
+  then
+    rm -f $dst_file
+  fi
+  for name in VOC2007 VOC2012
+  do
+    if [[ $dataset == "test" && $name == "VOC2012" ]]
+    then
+      continue
+    fi
+    echo "Create list for $name $dataset..."
+    dataset_file=$root_dir/$name/$sub_dir/$dataset.txt
+
+    img_file=$bash_dir/$dataset"_img.txt"
+    cp $dataset_file $img_file
+    sed -i "s/^/$name\/JPEGImages\//g" $img_file
+    sed -i "s/$/.jpg/g" $img_file
+
+    label_file=$bash_dir/$dataset"_label.txt"
+    cp $dataset_file $label_file
+    sed -i "s/^/$name\/Annotations\//g" $label_file
+    sed -i "s/$/.xml/g" $label_file
+
+    paste -d' ' $img_file $label_file >> $dst_file
+
+    rm -f $label_file
+    rm -f $img_file
+  done
+
+  # Generate image name and size infomation.
+  if [ $dataset == "test" ]
+  then
+    $bash_dir/../../build/tools/get_image_size $root_dir $dst_file $bash_dir/$dataset"_name_size.txt"
+  fi
+
+  # Shuffle trainval file.
+  if [ $dataset == "trainval" ]
+  then
+    rand_file=$dst_file.random
+    cat $dst_file | perl -MList::Util=shuffle -e 'print shuffle(<STDIN>);' > $rand_file
+    mv $rand_file $dst_file
+  fi
+done

data/VOC0712/create_list_fisheye.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+
+root_dir="../../data/VOCdevkit"
+sub_dir=ImageSets/Main
+bash_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+for dataset in trainval test
+do
+  dst_file=$bash_dir/$dataset.txt
+  if [ -f $dst_file ]
+  then
+    rm -f $dst_file
+  fi
+  for name in fisheye #VOC2007 VOC2012
+  do
+    if [[ $dataset == "test" && $name == "VOC2012" ]]
+    then
+      continue
+    fi
+    echo "Create list for $name $dataset..."
+    dataset_file=$root_dir/$name/$sub_dir/$dataset.txt
+
+    img_file=$bash_dir/$dataset"_img.txt"
+    cp $dataset_file $img_file
+    sed -i "s/^/$name\/JPEGImages\//g" $img_file
+    sed -i "s/$/.jpg/g" $img_file
+
+    label_file=$bash_dir/$dataset"_label.txt"
+    cp $dataset_file $label_file
+    sed -i "s/^/$name\/Annotations\//g" $label_file
+    sed -i "s/$/.xml/g" $label_file
+
+    paste -d' ' $img_file $label_file >> $dst_file
+
+    rm -f $label_file
+    rm -f $img_file
+  done
+
+  # Generate image name and size infomation.
+  if [ $dataset == "test" ]
+  then
+    $bash_dir/../../build/tools/get_image_size $root_dir $dst_file $bash_dir/$dataset"_name_size.txt"
+  fi
+
+  # Shuffle trainval file.
+  if [ $dataset == "trainval" ]
+  then
+    rand_file=$dst_file.random
+    cat $dst_file | perl -MList::Util=shuffle -e 'print shuffle(<STDIN>);' > $rand_file
+    mv $rand_file $dst_file
+  fi
+done

data/VOC0712/detect_test.txt

@@ -0,0 +1,5 @@
+C:\private\000166.jpg
+C:\private\001852.jpg
+C:\private\002597.jpg
+C:\private\004030.jpg
+C:\private\005225.jpg

data/VOC0712/labelmap_fisheye.prototxt

@@ -0,0 +1,45 @@
+item {
+  name: "none_of_the_above"
+  label: 0
+  display_name: "background"
+}
+item {
+  name: "g"
+  label: 1
+  display_name: "g"
+}
+item {
+  name: "l"
+  label: 2
+  display_name: "l"
+}
+item {
+  name: "b"
+  label: 3
+  display_name: "b"
+}
+item {
+  name: "c"
+  label: 4
+  display_name: "c"
+}
+item {
+  name: "m"
+  label: 5
+  display_name: "m"
+}
+item {
+  name: "y"
+  label: 6
+  display_name: "y"
+}
+item {
+  name: "p"
+  label: 7
+  display_name: "p"
+}
+item {
+  name: "t"
+  label: 8
+  display_name: "t"
+}

data/VOC0712/labelmap_voc.prototxt

@@ -0,0 +1,105 @@
+item {
+  name: "none_of_the_above"
+  label: 0
+  display_name: "background"
+}
+item {
+  name: "aeroplane"
+  label: 1
+  display_name: "aeroplane"
+}
+item {
+  name: "bicycle"
+  label: 2
+  display_name: "bicycle"
+}
+item {
+  name: "bird"
+  label: 3
+  display_name: "bird"
+}
+item {
+  name: "boat"
+  label: 4
+  display_name: "boat"
+}
+item {
+  name: "bottle"
+  label: 5
+  display_name: "bottle"
+}
+item {
+  name: "bus"
+  label: 6
+  display_name: "bus"
+}
+item {
+  name: "car"
+  label: 7
+  display_name: "car"
+}
+item {
+  name: "cat"
+  label: 8
+  display_name: "cat"
+}
+item {
+  name: "chair"
+  label: 9
+  display_name: "chair"
+}
+item {
+  name: "cow"
+  label: 10
+  display_name: "cow"
+}
+item {
+  name: "diningtable"
+  label: 11
+  display_name: "diningtable"
+}
+item {
+  name: "dog"
+  label: 12
+  display_name: "dog"
+}
+item {
+  name: "horse"
+  label: 13
+  display_name: "horse"
+}
+item {
+  name: "motorbike"
+  label: 14
+  display_name: "motorbike"
+}
+item {
+  name: "person"
+  label: 15
+  display_name: "person"
+}
+item {
+  name: "pottedplant"
+  label: 16
+  display_name: "pottedplant"
+}
+item {
+  name: "sheep"
+  label: 17
+  display_name: "sheep"
+}
+item {
+  name: "sofa"
+  label: 18
+  display_name: "sofa"
+}
+item {
+  name: "train"
+  label: 19
+  display_name: "train"
+}
+item {
+  name: "tvmonitor"
+  label: 20
+  display_name: "tvmonitor"
+}