GitHub
  
  
    

longcw/faster_rcnn_pytorch 48

faster_rcnn_pytorch - Faster RCNN with PyTorch


  
  
    
    
  
  


Thiss is an implementation of faster rcnn using pytorch. He wrote C++(CUDA) code for roi-pooling and non maximal suppression(NMS).
http://pytorch.org/docs/master/notes/extending.html 58
Pytorch docs for extending  torch.nn, torch.autograd, and writing custom C extensions utilizing our C libraries.
Note: If you figure out how to write cuda code and interface it with pytorch, please tell me.