search.sh has error

[akihiro0000]

Prepare the environment according to the procedure, I try "sh search.sh" in the Architecture Search 1st steps.

Traceback (most recent call last):
  File "/home/test/anaconda3/envs/leastereo/lib/python3.8/site-packages/PIL/Image.py", line 2813, in fromarray
    mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 192), '|u1')

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "search.py", line 303, in <module>
    trainer.training(epoch)
  File "search.py", line 211, in training
    self.summary.visualize_image_stereo(self.writer, input1, target, output, global_step)
  File "/home/test/test/LEAstereo/utils/summaries.py", line 56, in visualize_image_stereo
    writer.add_image('Predicted disparity', pr_image, global_step)
  File "/home/test/anaconda3/envs/leastereo/lib/python3.8/site-packages/tensorboardX/writer.py", line 667, in add_image
    image(tag, img_tensor, dataformats=dataformats), global_step, walltime)
  File "/home/test/anaconda3/envs/leastereo/lib/python3.8/site-packages/tensorboardX/summary.py", line 288, in image
    image = make_image(tensor, rescale=rescale)
  File "/home/test/anaconda3/envs/leastereo/lib/python3.8/site-packages/tensorboardX/summary.py", line 330, in make_image
    image = Image.fromarray(tensor)
  File "/home/test/anaconda3/envs/leastereo/lib/python3.8/site-packages/PIL/Image.py", line 2815, in fromarray
    raise TypeError("Cannot handle this data type: %s, %s" % typekey) from e
TypeError: Cannot handle this data type: (1, 1, 192), |u1

Do you have any idea?

[natalieschreder]

I'm running into this issue too! I know it's been a while, but did you ever figure out what the problem was?