Giters

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

Tensorflow implementation of "Semantic Instance Segmentation with a Discriminative Loss Function"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

error info

yghcats opened this issue · comments

commented

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)]]