https://github.com/ruimashita/caffe-train
Raw File
Tip revision: 0d955f939b451c84aee0c9df2503273034279233 authored by Takuya Wakisaka on 28 July 2016, 02:11:27 UTC
create imagenet db
Tip revision: 0d955f9
create_leveldb.sh
#!/usr/bin/env sh
DATA=/home/docker/data_set

echo "Creating leveldb..."

rm -rf $DATA/train_leveldb
rm -rf $DATA/val_leveldb

convert_imageset \
    -shuffle \
    -backend leveldb \
    -resize_height 256 \
    -resize_width 256 \
    $DATA/train/ \
    $DATA/train.txt \
    $DATA/train_leveldb 

convert_imageset \
    -shuffle \
    -backend leveldb \
    -resize_height 256 \
    -resize_width 256 \
    $DATA/val/ \
    $DATA/val.txt \
    $DATA/val_leveldb 

# convert_imageset \
#     -shuffle \
#     -resize_height 256 \
#     -resize_width 256 \
#     $DATA/train/ \
#     $DATA/train.txt \
#     $DATA/train_leveldb 

# convert_imageset \
#     -shuffle \
#     -resize_height 256 \
#     -resize_width 256 \
#     $DATA/val/ \
#     $DATA/val.txt \
#     $DATA/val_leveldb 


echo "Done."
back to top