Did anyone try this pytorch implementation and get reasonable results?

[gdwei]

trafficstars

Can this reposity reproduce the results in their paper? Or anyone who tries on their own datasets and get some similar results like those in the paper? I am not using these code yet, hope to get some advise from those who have tried it.

[SteveVanWang]

the pytorch model I have trianed for 5days with gtx2080ti makes an unreasonable result.hope someone can share a good pretrained model.

[gdwei]

the pytorch model I have trianed for 5days with gtx2080ti makes an unreasonable result.hope someone can share a good pretrained model.

Thank you for this information. I have trained on my own dataset and the results were not reasonable. But my data is not RBG image, so this info may not be helpful for those using image dataset.

[knicholes]

I'm using pytorch right now, and after just one day on my RTX 3090, my results are very promising!