BVLC / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
33.96k stars 18.72k forks source link

Segmentation fault (core dumped) when creating imageset #7055

Open SimonWang2014 opened 2 years ago

SimonWang2014 commented 2 years ago

Issue summary

Failed to create my own image set, Segmentation fault (core dumped).

Steps to reproduce

create_imagenet.sh config:

set -e

EXAMPLE=data
DATA=/scratch/Teaching/hw01220/dogs/NewImages
TOOLS=/opt/caffe/build/tools

TRAIN_DATA_ROOT=/scratch/Teaching/hw01220/dogs/NewImages/train_img/
VAL_DATA_ROOT=/scratch/Teaching/hw01220/dogs/NewImages/val_img/

....

GLOG_logtostderr=1 $TOOLS/convert_imageset \
    --resize_height=$RESIZE_HEIGHT \
    --resize_width=$RESIZE_WIDTH \
    --shuffle \
    $TRAIN_DATA_ROOT \
    $DATA/train.txt \
    $EXAMPLE/cars_train_lmdb

echo "Creating val lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
    --resize_height=$RESIZE_HEIGHT \
    --resize_width=$RESIZE_WIDTH \
    --shuffle \
    $VAL_DATA_ROOT \
    $DATA/val.txt \
    $EXAMPLE/cars_val_lmdb

The format of the train.txt and val.txt:

63_scratch6.jpg 63

output:

(22:09 hw01220@manfred imagenet) > ./create_imagenet.sh 
Creating train lmdb...
I0518 22:09:31.906777 2695928 convert_imageset.cpp:86] Shuffling data
I0518 22:09:32.294107 2695928 convert_imageset.cpp:89] A total of 17493 images.
I0518 22:09:32.348204 2695928 db_lmdb.cpp:35] Opened lmdb data/cars_train_lmdb
Segmentation fault (core dumped)

System configuration