lucent icon indicating copy to clipboard operation
lucent copied to clipboard

handle single channel images in tensor_to_img

Open alik-git opened this issue 2 years ago • 0 comments

If a custom model is being used where the optimized parameter image only has one channel, then the tensor_to_img_array function in render.py crashes.

By handling that possibility, the error is fixed.

alik-git avatar Nov 24 '23 01:11 alik-git