search

LiWentomng / Point2Mask

The code for "Point2Mask: Point-supervised Panoptic Segmentation via Optimal Transport", ICCV2023
Apache License 2.0
63 stars 6 forks source link

The loss for weakly supervised learning? #4

Open lzl2040 opened 10 months ago

lzl2040 commented 10 months ago

I can not find the code about the loss for weakly supervised learning. Could you tell me how to find them?

LiWentomng commented 10 months ago

@lzl2040 Hello, the main codes lie in point2mask_head.py. The code of loss for semantic map learning can be found from Line625 to Line635, and the loss for boundary map learning can be found from Line712 to Line769.