How to setting dataset_root

[geminixiang]

Hi~ Please help, I am in trouble... I used to run download.sh to download dataset, then unzip it (got folder 'picture_files' & 'coordinate_files'), and put in this project's root. But I have no idea to edit dataset_root in cnn_rotation.py,

2. Set dataset path or other settings in 'cnn_rotation.py'.
dataset_root = ${CUSTOMIZED_IMAGE_PRETRAIN_ROOT}

I've try fix it by

_sketch_img_root = './picture_files'
or
_sketch_img_root = './picture_files'

and use this command to run it

python train_cnn.py  --exp cnn_rotation  --num_workers 8 --gpu 0
but error
ValueError: num_samples should be a positive integer value, but got num_samples=0

which folder that it's pointed to is correct? or I got it wrong.

[improving-bookworm]

I met the same problem as you. Have you got it fixed yet?

[taner7]

I have the same problem, did you manage to solve it?