search

tensorlayer / SRGAN

Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
https://github.com/tensorlayer/tensorlayerx
3.28k stars 809 forks source link

Need help, I'm a newer #45

Open ydzhou-pku opened 6 years ago

ydzhou-pku commented 6 years ago

Please help me to handle this, I just dont know how to do with this. I use the same dataset(DIV2K..) with author, and didnt change anything of any program. But it just cant run...

errorTraceback (most recent call last): File "main.py", line 316, in train() File "main.py", line 73, in train net_g = SRGAN_g(t_image, is_train=True, reuse=False) File "/home/zhangheng_li/SRGAN/model.py", line 42, in SRGAN_g n = SubpixelConv2d(n, scale=2, n_out_channel=None, act=tf.nn.relu, name='pixelshufflerx2/1') File "/home/zhangheng_li/SRGAN/tensorlayer/layers.py", line 2193, in SubpixelConv2d net_new.outputs = act(_PS(inputs, r=scale, n_out_channel=n_out_channel)) File "/home/zhangheng_li/SRGAN/tensorlayer/layers.py", line 2175, in _PS Xs=tf.split(X,r,3) #bhwrr File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/array_ops.py", line 1234, in split name=name) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_array_ops.py", line 3241, in _split num_split=num_split, name=name) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/op_def_library.py", line 508, in apply_op (prefix, dtypes.as_dtype(input_arg.type).name)) TypeError: Input 'split_dim' of 'Split' Op has type float32 that does not match expected type of int32.

zsdonghao commented 6 years ago

what is the TF and TL version are you using?

ydzhou-pku commented 6 years ago

Really appreciate your reply firstly. I guess its 0.8.0, our server just reloaded two weeks ago, and I just gave up that. I'll redo that in a few weeks, if any question, I'll request for help again. REALLY THANKS :)