PyTorch example with CNN-based object detection

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.

Any help is greatly appreciated,
Plamen

3 Likes

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:


https://github.com/longcw/yolo2-pytorch
https://github.com/amdegroot/ssd.pytorch
7 Likes

Hi @fmassa,
Thanks for the great references.
Is there an intent to release any unified pytorch API/project for object detection ?

This is done recently by tensorflow (https://research.googleblog.com/2017/06/supercharge-your-computer-vision-models.html?m=1).
This kind of API can help reasearchers to easily :

  • switch between meta-architectures (SSD/Faster-RCNN) and base architectures (Resnet/Inception/…)
  • test training tricks such as hard-negatives mining
  • many other tweaks relevant for object detection

@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.

2 Likes

citing few more (prolly some better too)

YOLOv2:

Below one is Python3

FasterRCNN

Best place to check for more or other pytorch models would be soumith’s starred repos :stuck_out_tongue:

4 Likes

:stuck_out_tongue_closed_eyes::stuck_out_tongue_closed_eyes:

4 Likes

Hijacking the thread : is anyone aware of a pytorch implementation , or an easy way to port to pytorch the models in https://github.com/facebookresearch/Detectron ?

1 Like

@Amir_Rosenfeld

I started working on that. Check here https://github.com/ignacio-rocco/detectorch

1 Like

@ignacio-rocco Very cool, thanks for letting me know!