ndrplz / dilation-tensorflow Goto Github PK

hi @ndrplz,

when i replaced the "cityscapes.png" file in the data directory by some other test image with smaller resolution (e.g. 1438 x 803), then I got following error:

File "main_tf.py", line 61, in <module>
    predicted_image = predict(input_image, input_tensor, model, dataset, sess)
  File "C:\Users\ininetynine\Desktop\dilation-tensorflow-master\utils.py", line 79, in predict
    color_image = CONFIG[ds]['palette'][prediction.ravel()].reshape(image_size)
ValueError: total size of new array must be unchanged

any idea on the reason behind this? and how should i fix it? thank you ~

pretrained_dilation_camvid.pickle and pretrained_dilation_cityscapes.pickle，How does this come about？How to experiment with my own data set?

I ran into a ResourceExhaustedError errror on my Jetson TX1 with 4GB of RAM. How much RAM is needed/recommended to test the code???

Ubuntu 16.04
python 3.5
tensorflow 1.5.0
cv2 3.3.0
pretrained_dilation_cityscapes.pickle - 537MB

ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape [7,7,512,4096] and type float
[[Node: fc6/Variable/initial_value = Const[dtype=DT_FLOAT, value=Tensor<type: float shape: [7,7,512,4096] values: [[[-4.46356426e-05 0.000302301254 -0.00146040611]]]...>, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]

main_tf.py with pre-trained Cityscapes weights works with no problems when I run it on the dummy image under /data.
However when I ran it on my own image here, I encountered the following error:

Traceback (most recent call last):
  File "main_tf.py", line 60, in <module>
    predicted_image = predict(input_image, input_tensor, model, dataset, sess)
  File "dilation-tensorflow/utils.py", line 76, in predict
    color_image = CONFIG[ds]['palette'][prediction.ravel()].reshape(image_size)
ValueError: cannot reshape array of size 50331648 into shape (3183,3183,3)

50331648 corresponds to a shape of (4096,4096,3) which is not the image size. What might be the issue here?