How can I visualize free mask results?

[STAR-811]

To ensure that the free masks are almost accurate, I add following codes in inference_freemaks.py . However, the results saved in the disks is strange: the foreground part is always at the edge of an image.

after masks = masks.cpu().numpy() ''' masks = masks.cpu().numpy() ii=0 for mask in masks: masks_save = masks[ii] * 255 cv2.imwrite('./results/'+img_name+'_'+str(ii)+'.jpg', masks_save) ii=ii+1 '''

[ahuirecome]

same with you

[Wayne2Wang]

I had the same issue when I was using resnet50. Switching to resnet101 worked.