timy90022 / One-Shot-Object-Detection

Implementation of One-Shot Object Detection with Co-Attention and Co-Excitation in Pytorch
MIT License
423 stars 83 forks source link

Non local block operation #21

Open prathmeshrmadhu opened 4 years ago

prathmeshrmadhu commented 4 years ago

Hi,

Can anyone point me specifically about where can I find the non-local block which the authors have mentioned as the first step in their paper (Section 3) ? I am having a hard time finding the same in the code. Thanks.

Best, Prathmesh.

IISCAditayTripathi commented 4 years ago

In faster_rcnn.py there is class called matchnet. This is where non-local block is implemented.