search

Leonardo-Blanger / detr_tensorflow

A Tensorflow implementation of the DETR object detection architecture.
MIT License
62 stars 15 forks source link

Can you please show hungarian loss implementation of pytorch in tensorflow. #7

Open Gruhit13 opened 2 years ago

Gruhit13 commented 2 years ago

I am also ML enthusiast who came through this paper and was willing to replicate this entire behavior in tensorflow. Fortunately I even completed the project but due to lack of my knowledge I am not able to put this last piece which resides at the heart of DETR and that is implementation of its loss calculation. I tried a lot of ways but that particular this is one which I am not able to implement using tensorflow. So if you can help me with that then it would be so great of you.

You can contact me on my email if you want to see my implementation. My email id is: gruhitml13@gmail.com

LucasKirsten commented 2 years ago

Hi, I have implemented the Hungarian Loss in here: https://github.com/LucasKirsten/detr_tf2/blob/master/misc/losses.py#L76

Gruhit13 commented 2 years ago

Thank you so much sir for replying. I think my current knowledge is not such that I can understand the code. As I am having many doubts which I need to answer. Thanks once again for sharing your code with me.

On Tue, 22 Mar 2022 at 06:15, Lucas Kirsten @.***> wrote:

Hi, I have implemented the Hungarian Loss in here: https://github.com/LucasKirsten/detr_tf2/blob/master/misc/losses.py#L76

— Reply to this email directly, view it on GitHub https://github.com/Leonardo-Blanger/detr_tensorflow/issues/7#issuecomment-1074563459, or unsubscribe https://github.com/notifications/unsubscribe-auth/APJEH4YCQ66CEIWCBUZ6AETVBEJ37ANCNFSM5RHFOFFA . You are receiving this because you authored the thread.Message ID: @.***>