uber-research / UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Other
649 stars 120 forks source link

Question about visualization when trying to test COCO dataset(train2014) #21

Closed AllenSpurs closed 5 years ago

AllenSpurs commented 5 years ago

Thanks for your great work! I had trained the model with COCO train2017&val2017, and pulled the latest code, but when I trying to test COCO dataset (train2014) with setting the vis_mask=True, I still only got an instance segmentation like figure. Is there anything that needs special attention? COCO_train2014_000000000025 jpg

dingguo1996 commented 5 years ago

Do you want to get the panoptic segmentation figure?

AllenSpurs commented 5 years ago

Yes, that is what I want, I just want to utilize the panoptic segmentation figure to extract some image feature. Need any sepcific setting here? Thank you.

dingguo1996 commented 5 years ago

No, it will produce the panoptic segmentation figures for you automatically!

AllenSpurs commented 5 years ago

So if all goes well, should the panoptic segmentation figures be in the directory like "UPSNet/output/upsnet/coco/upsnet_resnet50_coco_4gpu/test2015/results/vis"?

dingguo1996 commented 5 years ago

directory like "UPSNet/upsnet/output/upsnet/coco/upsnet_resnet50_coco_4gpu/val2017/results/pans_unified/pan/"

AllenSpurs commented 5 years ago

OK, I will set the mask_vis to False and try it again. Thank you for your patient reply.