search

aurora95 / Keras-FCN

Keras-tensorflow implementation of Fully Convolutional Networks for Semantic Segmentation（Unfinished）
MIT License
650 stars 268 forks source link

ValueError: axes don't match array #28

Open mahemon opened 7 years ago

mahemon commented 7 years ago

Hi @aurora95 and @athundt, When I run the train.py file, it raises the following error. Could you please suggest me some way to solve this issue.

File "train.py", line 213, in label_suffix=label_suffix, ignore_label=ignore_label, label_cval=label_cval) File "train.py", line 83, in train classes=classes) ValueError: axes don't match array

if dataset == 'VOC2012_BERKELEY': train_file_path = os.path.expanduser('~/tf/keras-fcn/datasets/VOC2012/ImageSets/Segmentation/train.txt') val_file_path = os.path.expanduser('~/tf/keras-fcn/datasets/VOC2012/ImageSets/Segmentation/val.txt') data_dir = os.path.expanduser('~/tf/keras-fcn/datasets/VOC2012/JPEGImages') label_dir = os.path.expanduser('~/tf/keras-fcn/datasets/VOC2012/SegmentationClass') data_suffix='.jpg' label_suffix='.png' classes = 21

zxpeter commented 5 years ago

Same issue...When change 'model_name = 'FCN_Vgg16_32s''

songyu3044 commented 5 years ago

I had the same problem,When change 'model_name = 'FCN_Vgg16_32s''，please help me