gj-nvd-diffusion

Implementation of "CNN Combined With a Prior Knowledge-based Candidate Search and Diffusion Method for Nighttime Vehicle Detection"
paper link

Log

20240522: Added missing data augmentation files. Updated candidate scale adjustment.
20240523: In the published paper, matching between candidate and gt is introduced as follows: The 1:1 matching of candidates and GTs is performed using bipartite matching; However, 1:5 matching was being performed. It's my fault ...

Weights

Google drive link gives the pre-trained weights of our network.
google drive link

Setup

First, install an older version of pytorch. (not over 2.0.0)
And follow the instructions below to install Detectron2.
Detectron2 link
The pytorch version I tested on is 1.12.0 and the Detectron2 version is 0.6.
Please modify the path of the config file and the path of the main file.
I provide a small data set for testing.
I have prepared 100 images and annotations in the coco data set format in the ./gendata folder.

Run

python train_net.py --num-gpus 1 --config-file configs/diffnvd.park.best.yaml --eval-only MODEL.WEIGHTS gendata/best.pth

Demo

I'm working on demo code. SORRY.

Problems

Any questions regarding my published papers and code are always welcome.

Special Thanks

DiffusionDet: Diffusion Model for Object Detection
스마트 전조등을 위한 비전 기반 야간 차량 인식

GwangJu Dataset sharing (Koreans Only)

데이터 셋의 용량이 약 100Gb(스테레오영상 약 30,000장)가 넘어서 저의 능력으로는 온라인 공유가 어렵습니다.
만약 광주 데이터 셋이 연구에 필요하시면 제가 전체 데이터 셋을 물리적인 방법(전남대학교 방문 or 출장)으로 드릴 수 있습니다.
공유에 대한 요청은 언제든지 환영입니다. 감사합니다.