GeorgeSeif / Semantic-Segmentation-Suite

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
2.5k stars 880 forks source link

how do I use input data with shape (width, height, 2) or 4 channels? #180

Open apple1986 opened 5 years ago

apple1986 commented 5 years ago

Hello, thanks for your code. When I input my data with two or 4 channels, it shows the questions as follows: can you tell me how to deal with this problem. I am very appreciated you....

InvalidArgumentError (see above for traceback): Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Assign requires shapes of both tensors to match. lhs shape= [7,7,2,64] rhs shape= [7,7,3,64] [[Node: save/Assign_249 = Assign[T=DT_FLOAT, _class=["loc:@resnet_v2_50/conv1/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](resnet_v2_50/conv1/weights, save/RestoreV2/_499)]]