racinmat / anime-style-transfer

Style transfer for between real photos and anime images using CycleGAN
55 stars 8 forks source link

'import/real-anime-gen/c-7-1-64-r/Conv2D' (op: 'Conv2D') with input shapes: [1,?,512,512,?], [7,7,3,64] #4

Open Harryjun opened 4 years ago

Harryjun commented 4 years ago

I have some problems like this

Traceback (most recent call last):
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/tensorflow/python/framework/importer.py", line 427, in import_graph_def
    graph._c_graph, serialized, options)  # pylint: disable=protected-access
tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be rank 4 but is rank 5 for 'import/real-anime-gen/c-7-1-64-r/Conv2D' (op: 'Conv2D') with input shapes: [1,?,512,512,?], [7,7,3,64].

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "transform.py", line 175, in <module>
    tf.app.run()
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/tensorflow/python/platform/app.py", line 40, in run
    _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/absl/app.py", line 299, in run
    _run_main(main, args)
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/absl/app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "transform.py", line 153, in main
    transform_files(im_paths)
  File "transform.py", line 137, in transform_files
    feeder, ceil(len(im_paths) / FLAGS.batchsize), FLAGS.batchsize, persist_images_postprocessing)
  File "/home/evga/work/mlj/mlj-demo/face_dongman/anime-style-transfer/code/mod-cycle-gan/cycle/_model.py", line 398, in test_one_part
    d_input, d_output, output = CycleGAN.get_graph_outputs(graph, dataset[0], pb_model)
  File "/home/evga/work/mlj/mlj-demo/face_dongman/anime-style-transfer/code/mod-cycle-gan/cycle/_model.py", line 430, in get_graph_outputs
    input_map={'input': input_var},return_elements=CycleGAN.OUTPUT_NODES)))
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/home/evga/anaconda3/envs/mlj7/lib/python3.7/site-packages/tensorflow/python/framework/importer.py", line 431, in import_graph_def
    raise ValueError(str(e))
ValueError: Shape must be rank 4 but is rank 5 for 'import/real-anime-gen/c-7-1-64-r/Conv2D' (op: 'Conv2D') with input shapes: [1,?,512,512,?], [7,7,3,64].

can anyone give me some answer.