search

pse-ecn / pose-sensitive-embedding

Pose Sensitive Embedding for Person Re-Identification (PSE)
112 stars 36 forks source link

I encounter "InvalidArgumentError" when I predict the model #15

Closed THHHomas closed 6 years ago

THHHomas commented 6 years ago

I download the pretrained model and use the command "python predictor_preid.py --model-dir=./models --data=./Data/Market-1501-v15.09.15 --dataset-name=market1501 --batch-size=128 --network-name=resnet_v1_50_views" more, I write my checkpoint file like this: model_checkpoint_path: "model.ckpt-104801" all_model_checkpoint_paths: "model.ckpt-104801"

and i encounter the following error: tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [7,7,3,64] rhs shape= [7,7,17,64] [[Node: save/Assign_382 = Assign[T=DT_FLOAT, _class=["loc:@resnet_v1_50/conv1/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](resnet_v1_50/conv1/weights, save/RestoreV2/_777)]]

cppntn commented 5 years ago

@THHHomas hello, I am facing the same error! How did you overcome it?

THHHomas commented 5 years ago

@antocapp , you need to get the pose map which is 17 channels feature before you use this program, please refer to https://github.com/pse-ecn/pose-sensitive-embedding/issues/13