Joker316701882 / Deep-Image-Matting

This is tensorflow implementation for paper "Deep Image Matting"
625 stars 188 forks source link

help!help!help!Test run failed #23

Open PanJinquan opened 6 years ago

PanJinquan commented 6 years ago

I download your Github code and the corresponding model file:https://github.com/Joker316701882/Deep-Image-Matting,Then run:python test.py --alpha=./test_data/alpha/1.png --rgb=./test_data/RGB/1.png,But error in this procedure:saver.restore(sess,tf.train.latest_checkpoint('./model'))#PS:I have downloaded your model and put it in the "model" file. Here errors:InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels:

[[Node: pool1 = MaxPoolWithArgmax[T=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]](conv1_2)]]
fpipart commented 6 years ago

This function 'MaxPoolWithArgmax' runs only on GPU on older version of tensorflow: https://github.com/tensorflow/tensorflow/issues/6035

It should work on CPU with tensorflow 1.8

elzeru commented 6 years ago

Just tried 1.8 CPU, works like a charm, and the upgrade is very easy to do. On my Debian: pip3 install --upgrade tensorflow

shliang0603 commented 6 years ago

@PanJinquan Can you send me the pretrained model, I can't download it ,my email is :shliang0603@gmail.com Thanks

shliang0603 commented 6 years ago

@PanJinquan 哥哥,你有Deep-Image-Matting的数据集吗,可以发我一份吗,谢谢啦 ,我的email: shliang0603@gmail.com

chenhaocs commented 5 years ago

I download your Github code and the corresponding model file:https://github.com/Joker316701882/Deep-Image-Matting,Then run:python test.py --alpha=./test_data/alpha/1.png --rgb=./test_data/RGB/1.png,But error in this procedure:saver.restore(sess,tf.train.latest_checkpoint('./model'))#PS:I have downloaded your model and put it in the "model" file. Here errors:InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels: [[Node: pool1 = MaxPoolWithArgmaxT=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]]]

@PanJinquan Hello, could you send me your pre-trained model and the training datasets? Thanks a lot! My email chanhal@outlook.com

rainsun1 commented 5 years ago

@PanJinquan could you ALSO send me your pre-trained model and the training datasets? Thanks! email: yglhit@sina.cn

ssxlconch commented 5 years ago

I download your Github code and the corresponding model file:https://github.com/Joker316701882/Deep-Image-Matting,Then run:python test.py --alpha=./test_data/alpha/1.png --rgb=./test_data/RGB/1.png,But error in this procedure:saver.restore(sess,tf.train.latest_checkpoint('./model'))#PS:I have downloaded your model and put it in the "model" file. Here errors:InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels: [[Node: pool1 = MaxPoolWithArgmaxT=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]]]

sorry for bothering, would you please send me the pretrained-model? my emai: 2741592109@qq.com