IGEV icon indicating copy to clipboard operation
IGEV copied to clipboard

Need help to do inference for gray scale image

Open mkothule opened this issue 1 year ago • 4 comments

Want to run the network for gray scale images (single channel)

I get this error while running network on gray images Traceback (most recent call last): File "demo_imgs.py", line 100, in demo(args) File "demo_imgs.py", line 50, in demo image1 = load_image(imfile1) File "demo_imgs.py", line 29, in load_image img = torch.from_numpy(img).permute(2, 0, 1).float() RuntimeError: number of dims don't match in permute

I tried copying same gray values for all 3 channel but results are not very good.

I see eth3d is gray scale image dataset so I also tried with eth3d network shared. But I still get above error.

Can you please share what change is needed to adapt network to gray images?

mkothule avatar Aug 10 '23 15:08 mkothule

can you give me your gray images?

gangweiX avatar Aug 11 '23 02:08 gangweiX

Currently I am using KITTIT images (RGB2Gray converted) for experimentation. 000000_10_image_3 png_gray 000000_10_image_2_gray

mkothule avatar Aug 11 '23 15:08 mkothule

You can use KITTI pretrained model, that will perform well.

gangweiX avatar Aug 12 '23 00:08 gangweiX

thanks gangweiX. I see sensible output with kitti2015 pre-trained network for above images.

mkothule avatar Aug 14 '23 12:08 mkothule