ValueError: Dimension 0 in both shapes must be equal, but are 3 and 64. Shapes are [3,3,18,64] and [64,21,3,3]. for 'Assign_2' (op: 'Assign') with input shapes: [3,3,18,64], [64,21,3,3].

[TwistSun]

Hi,
when i use your checkpoint file "generator-warp-mask-nn3-cl12.h5" for market test, I get this error:
Using TensorFlow backend.
2019-10-20 16:21:09.954087: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
Generate images...
Number of images: 32668
Number of pairs train: 263632
Number of pairs test: 12000
WARNING:tensorflow:From /data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/keras/backend/tensorflow_backend.py:1290: calling reduce_mean (from tensorflow.python.ops.math_ops) with keep_dims is deprecated and will be removed in a future version.
Instructions for updating:
keep_dims is deprecated, use keepdims instead
Traceback (most recent call last):
File "/data00/home/menyifang/code/comparison/pose-gan-fix/pose-gan/test.py", line 152, in
test()
File "/data00/home/menyifang/code/comparison/pose-gan-fix/pose-gan/test.py", line 118, in test
generator.load_weights(args.generator_checkpoint)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/keras/engine/topology.py", line 2619, in load_weights
load_weights_from_hdf5_group(f, self.layers)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/keras/engine/topology.py", line 3095, in load_weights_from_hdf5_group
K.batch_set_value(weight_value_tuples)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/keras/backend/tensorflow_backend.py", line 2188, in batch_set_value
assign_op = x.assign(assign_placeholder)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/ops/variables.py", line 594, in assign
return state_ops.assign(self._variable, value, use_locking=use_locking)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/ops/state_ops.py", line 276, in assign
validate_shape=validate_shape)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/ops/gen_state_ops.py", line 59, in assign
use_locking=use_locking, name=name)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 3162, in create_op
compute_device=compute_device)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 3208, in _create_op_helper
set_shapes_for_outputs(op)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 2427, in set_shapes_for_outputs
return _set_shapes_for_outputs(op)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 2400, in _set_shapes_for_outputs
shapes = shape_func(op)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 2330, in call_with_requiring
return call_cpp_shape_fn(op, require_shape_fn=True)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/common_shapes.py", line 627, in call_cpp_shape_fn
require_shape_fn)
File "/data00/home/menyifang/anaconda3/envs/tfold/lib/python2.7/site-packages/tensorflow/python/framework/common_shapes.py", line 691, in _call_cpp_shape_fn_impl
raise ValueError(err.message)
ValueError: Dimension 0 in both shapes must be equal, but are 3 and 64. Shapes are [3,3,18,64] and [64,21,3,3]. for 'Assign_2' (op: 'Assign') with input shapes: [3,3,18,64], [64,21,3,3].

Do you have any idea to solve? (environment is as same as yours and it works well for fashion dataset) Thanks!

[AliaksandrSiarohin]

What is you image_data_format in ~/.keras/keras.json ?

[TwistSun]

What is you image_data_format in ~/.keras/keras.json ?
This is the content in my keras.json:
{
"floatx": "float32",
"epsilon": 1e-07,
"backend": "tensorflow",
"image_data_format": "channels_last"
}

[AliaksandrSiarohin]

Is there is the same problem for all the other market checkpoints?

[TwistSun]

Is there is the same problem for all the other market checkpoints?

Hi, It is ok for checkpoint file 'generator-warp-mask.h5'. Which checkpoint should I use to test full model in the paper?

[AliaksandrSiarohin]

generator-warp-mask-nn3-cl12.h5 is the correct one.
I just check it, run fine for me.
What command you use to run?