search

bowenc0221 / panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)
Apache License 2.0
587 stars 117 forks source link

Problem about visulizing the predict result #94

Closed guoxuerui closed 3 years ago

guoxuerui commented 3 years ago

After training the model on custom dataset, we want to visualize the predict result with trained weight. However, when use v.draw_panoptic_seg_predictions , the argument segments_info is not directly result from Panoptic-Deeplab prediction, the latter results contains three keys ['sem_seg', 'panoptic_seg', 'instances'], so we have no clue where the segements_info come from, is there any visulization tutorial we could follow?

bowenc0221 commented 3 years ago

You can pass panoptic_seg and set segments_info to None.