Hey Everyone, I have written a custom loss function (Focal Loss) in Pytorch which I am going to use in my SSD code.
Here is the colab URL for the loss function:- https://colab.research.google.com/drive/1NiUwI6ohF1b9TthutyfJQnOJ2i39R5cJ
I need peer reviews on my code regarding its correctness, I have done certain tests which are successful but not so sure whether it will work as intended for object detection is SSD.
Any feedback or suggestions for the code is extremely Appreciated. It would really help me out.