TimoSaemann / ENet

ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation
577 stars 276 forks source link

Training with Multichannel dataset #41

Open mtrth opened 6 years ago

mtrth commented 6 years ago

I am trying to train ENet with a multichannel dataset, input image size is 320x384. I get the following error on training, I1007 10:52:27.313614 10208 layer_factory.hpp:77] Creating layer conv5_0_4 I1007 10:52:27.313629 10208 net.cpp:100] Creating Layer conv5_0_4 I1007 10:52:27.313638 10208 net.cpp:434] conv5_0_4 <- prelu4_2_4_prelu4_2_4_0_split_1 I1007 10:52:27.313649 10208 net.cpp:408] conv5_0_4 -> conv5_0_4 I1007 10:52:27.313881 10208 net.cpp:150] Setting up conv5_0_4 I1007 10:52:27.313894 10208 net.cpp:157] Top shape: 3 16 96 80 (368640) I1007 10:52:27.313900 10208 net.cpp:165] Memory required for data: 971200512 I1007 10:52:27.313910 10208 layer_factory.hpp:77] Creating layer bn5_0_4 I1007 10:52:27.313928 10208 net.cpp:100] Creating Layer bn5_0_4 I1007 10:52:27.313937 10208 net.cpp:434] bn5_0_4 <- conv5_0_4 I1007 10:52:27.313947 10208 net.cpp:408] bn5_0_4 -> bn5_0_4 I1007 10:52:27.314162 10208 net.cpp:150] Setting up bn5_0_4 I1007 10:52:27.314174 10208 net.cpp:157] Top shape: 3 16 96 80 (368640) I1007 10:52:27.314180 10208 net.cpp:165] Memory required for data: 972675072 I1007 10:52:27.314190 10208 layer_factory.hpp:77] Creating layer upsample5_0_4 I1007 10:52:27.314200 10208 net.cpp:100] Creating Layer upsample5_0_4 I1007 10:52:27.314208 10208 net.cpp:434] upsample5_0_4 <- bn5_0_4 I1007 10:52:27.314218 10208 net.cpp:434] upsample5_0_4 <- pool1_0_4_mask I1007 10:52:27.314229 10208 net.cpp:408] upsample5_0_4 -> upsample5_0_4 I1007 10:52:27.314239 10208 upsample_layer.cpp:27] Params 'padout{}_' are deprecated. Please declare upsample height and width useing the upsample_h, upsample_w parameters. F1007 10:52:27.314256 10208 upsample_layer.cpp:58] Check failed: bottom[0]->channels() == bottom[1]->channels() (16 vs. 21)

I am trying to use the same architecture as your default prototxt, enet_train_encoder_decoder.prototxt. I am not sure why the upsample5_0_4 layer is expecting a 21 channel input.