Hi guys, I was wondering is there any example or at least pull request in progress regarding a PyTorch example with CNN-based object detection? For instance a short enough code on the COCO detection dataset? I looked in the examples on GitHub but at least I couldn’t find anything similar.
There was a pending PR in https://github.com/pytorch/examples/pull/21 that implemented a Faster R-CNN object detector in PyTorch, but I never finished testing it. Since then, several implementations were done, to cite a few:
@YossiB yes, it would definitely be valuable. A couple of years ago I released exactly the same thing for lua torch. Unfortunately, I don’t think I will have time anytime soon to do something similar now for pytorch, but one can definitely get lots of inspiration from the tensorflow implementation of object detection, which would make the task of implementing it simpler.