dice loss - Githubissues

abhi1kumar / SeaBird

[CVPR 2024] Official PyTorch Code of SeaBird: Segmentation in Bird's View with Dice Loss Improves Monocular 3D Detection of Large Objects

https://arxiv.org/pdf/2403.20318.pdf

MIT License

75 stars 5 forks source link

dice loss #4

Closed huzkun closed 2 months ago

huzkun commented 2 months ago

In which folder is dice loss located?

abhi1kumar commented 2 months ago

Hi @huzkun Thank you for your interest in SeaBird. The dice loss is implemented here in these two models:

PanopticBEV (KITTI-360 Single Camera Models): https://github.com/abhi1kumar/SeaBird/blob/0fd45183128f7d8615263e330484e2545004607c/PanopticBEV/panoptic_bev/algos/semantic_seg.py#L166-L176
HoP (nuScenes Multi-Camera Models): https://github.com/abhi1kumar/SeaBird/blob/0fd45183128f7d8615263e330484e2545004607c/HoP/mmdet3d_plugin/models/dense_heads/loss_utils.py#L22-L38

PS: It would be great if you could support our repo by starring it.

huzkun commented 2 months ago

Thanks for your answer, seabird is a great job.