Hi Miquel,
Thank you for sharing the scirpt. I have an issue when run it. I used the command
python create_dataset_lmdb.py --W 256 --H 256 --output_path /home/john/VOC2012ext_train_aug_label_lmdb --input_images /media/john/pascal/VOCdevkit/VOC2012_train_val_list/train_label_lmdb.txt --nb_classes 21 --labels
After that, I run the traning process, I got the error
softmax_loss_layer.cpp:42] Check failed: outer_num_ * inner_num_ == bottom[1]->count() (166500 vs. 65536) Number of labels must match number of predictions; e.g., if softmax axis == 1 and prediction shape is (N, C, H, W), label count (number of labels) must be N*H*W, with integer values in {0, 1, ..., C-1}.
I think the error is happened in converting the lmdb file. For more details, let see my issue at
https://groups.google.com/forum/#!topic/caffe-users/QdShjZtzY4Q . Thanks
Vào 13:08:22 UTC+9 Thứ Tư, ngày 28 tháng 12 năm 2016, Miquel Martí đã viết: