search

vanhuyz / CycleGAN-TensorFlow

An implementation of CycleGan using TensorFlow
MIT License
1.19k stars 436 forks source link

How to convert the ckpt model into savedmodel #126

Open Homura2333 opened 3 years ago

Homura2333 commented 3 years ago

In order to serve with tf-serving, the model needs to be converted into savedmodel. How to convert the ckpt model into savedmodel?

Homura2333 commented 3 years ago

Why does not the output op have shape?

I run the following code: print(graph.get_operation_by_name("input_image").outputs[0].shape) print(grpah.get_operation_by_name("output_image").outputs[0].shape)

and got the following results: (256,256,3) ()

So why is the output empty? This leads to the results that the converted saved model can not apply to triton inference server or tf-serving.