# Relation_Networks-pytorch

**Repository Path**: atari/Relation_Networks-pytorch

## Basic Information

- **Project Name**: Relation_Networks-pytorch
- **Description**: 同步 https://github.com/heefe92/Relation_Networks-pytorch
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-04-09
- **Last Updated**: 2022-03-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Relation_Networks-pytorch
Relation Networks for object detection based pytorch

### Progress

- [x] add relation module
- [x] add nms using relation module
- [x] end to end train renet based faster-rcnn on VOC2007
- [x] evaluate mAP
- [ ] modified learn_nms
- [ ] add OHEM
- [ ] improve performance
- [ ] train and forward on COCO

## Requirements
1. Python3.5
2. The following Python packages:
  ```
  Cython
  cupy
  numpy
  pytorch
  opencv-python
  collections
  ```
## Acknowledgement
This work builds on many excellent works, which include:

- [Yusuke Niitani's ChainerCV](https://github.com/chainer/chainercv) (mainly)
- [Ruotian Luo's pytorch-faster-rcnn](https://github.com/ruotianluo/pytorch-faster-rcnn) which based on [Xinlei Chen's tf-faster-rcnn](https://github.com/endernewton/tf-faster-rcnn)
- [simple-faster-rcnn-pytorch by chenyuntc](https://github.com/chenyuntc/simple-faster-rcnn-pytorch)
- [faster-rcnn.pytorch by Jianwei Yang and Jiasen Lu](https://github.com/jwyang/faster-rcnn.pytorch).It mainly refer to [longcw's faster_rcnn_pytorch](https://github.com/longcw/faster_rcnn_pytorch)
- All the above Repositories have referred to [py-faster-rcnn by Ross Girshick and Sean Bell](https://github.com/rbgirshick/py-faster-rcnn)  either directly or indirectly. 
- [yhenon's pytorch-retinanet](https://github.com/yhenon/pytorch-retinanet)
- [msracver's Relation-Networks-for-Object-Detection](https://github.com/msracver/Relation-Networks-for-Object-Detection)