vivian13maker / NLFFT-network

The source code will public soon

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

README

This repository implements NLFFTNet. The implementation is heavily influenced by the projects RFB, Detr. This repository aims to be the code base for researches based on RFB, Detr.

Installation

  1. Python3
  2. PyTorch 1.5+ or higher
  3. yacs
  4. tqdm
  5. scipy
  6. GCC >= 4.9
  7. OpenCV

Module

Data augment Module

The example using of CMS can be download.

CMS.py

NLFFT

The example of using of NLFFT Module can be download.

[Example of NLFFT.py](Example of NLFFT.py)

Citation

@article{ZENG202215, title = {NLFFTNet: A non-local feature fusion transformer network for multi-scale object detection}, journal = {Neurocomputing}, volume = {493}, pages = {15-27}, year = {2022}, issn = {0925-2312}, doi = {https://doi.org/10.1016/j.neucom.2022.04.062}}

About

The source code will public soon

License:MIT License


Languages

Language:Python 100.0%