How to implement on the origin mmdetection

hyz-xmaster / VarifocalNet

VarifocalNet: An IoU-aware Dense Object Detector

Apache License 2.0

348 stars 52 forks source link

How to implement on the origin mmdetection #5

Open FibonacciZ opened 3 years ago

FibonacciZ commented 3 years ago

I've installed mmdetection and I want to implement your algorithm，compared with the original mmdetection, does your implementation only need to add the following files： 1.configs/vfnet 2.models/dense_heads/vfnet_head.py 3.models/detectors/vfnet.py 4.models/detectors/varifocal_loss.py

hyz-xmaster commented 3 years ago

The answer is roughly Yes, but you also need to register these modules by adding them in corresponding init.py files.

But I think you do not need to implement it by yourself. Please see this pull request.