I am getting an error of KeyError: ((1, 1, 99), '|u1') with TypeError: Cannot handle this data type.
More specific error:
Epoch [1/101] training Loss: 1.8266
Traceback (most recent call last):
File "/home/akshita/anaconda2/envs/Faster_RCNN/lib/python3.6/site-packages/PIL/Image.py", line 2460, in fromarray
mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 99), '|u1')
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "patch_train.py", line 365, in <module>
train(args)
File "patch_train.py", line 204, in train
writer.add_image('train/original_label',correct_label_decoded, epoch + 1)
File "/home/akshita/anaconda2/envs/Faster_RCNN/lib/python3.6/site-packages/tensorboardX/writer.py", line 412, in add_image
self.file_writer.add_summary(image(tag, img_tensor), global_step, walltime)
File "/home/akshita/anaconda2/envs/Faster_RCNN/lib/python3.6/site-packages/tensorboardX/summary.py", line 205, in image
image = make_image(tensor, rescale=rescale)
File "/home/akshita/anaconda2/envs/Faster_RCNN/lib/python3.6/site-packages/tensorboardX/summary.py", line 243, in make_image
image = Image.fromarray(tensor)
File "/home/akshita/anaconda2/envs/Faster_RCNN/lib/python3.6/site-packages/PIL/Image.py", line 2463, in fromarray
raise TypeError("Cannot handle this data type")
TypeError: Cannot handle this data type``