search

hq-jiang / instance-segmentation-with-discriminative-loss-tensorflow

Tensorflow implementation of "Semantic Instance Segmentation with a Discriminative Loss Function"
MIT License
170 stars 47 forks source link

error info #9

Closed yghcats closed 5 years ago

yghcats commented 6 years ago

Hi, I run: python inference.py --modeldir trained_model --outdir inference_test/results What is the problem? tensorflow:1.4 thanks! error info: InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [2,2,1,16] rhs shape= [2,2,3,16] [[Node: save/Assign_538 = Assign[T=DT_FLOAT, _class=["loc:@Instance/transfer_layer/conv2d_transpose/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Instance/transfer_layer/conv2d_transpose/weights, save/RestoreV2_538/_3)]]