The definitions of sigmoid_focal_loss and dice_loss not found in model/detector.py

hustvl / YOLOS

[NeurIPS 2021] You Only Look at One Sequence

https://arxiv.org/abs/2106.00666

MIT License

827 stars 118 forks source link

The definitions of sigmoid_focal_loss and dice_loss not found in model/detector.py #20

Open Juanal-Xu opened 2 years ago

Juanal-Xu commented 2 years ago

Hi,

When I read your code, I found "Unresolved reference 'sigmoid_focal_loss' and Unresolved reference 'dice_loss' " in 'model/detector.py' file, and I could not find any code related to these two functions. May I ask if you forget to upload the definitions of the two functions?

Looking forward to your reply.