search

NVlabs / FreeSOLO

FreeSOLO for unsupervised instance segmentation, CVPR 2022
Other
312 stars 32 forks source link

How can I visualize free mask results? #6

Open STAR-811 opened 2 years ago

STAR-811 commented 2 years ago

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/'+imgname+''+str(ii)+'.jpg', masks_save) ii=ii+1 '''

ahuirecome commented 1 year ago

same with you

Wayne2Wang commented 1 year ago

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