Authors: Ruozhen He, Jiaying Lin, and Rynson Lau [Paper Link]
- We provide HetNet training weights on MSD and PMD presented in the papers.
- We provide HetNet testing maps on MSD and PMD presented in the papers.
- The pretrained weight can be found here: ResNeXt-101.
- Put the downloaded weights under
./resnext/
.
- Download the dataset and pretrained model.
- Modify the path in
train.py
. - Run
python train.py
.
- Modify the path and filename.
- Run
python test.py
. - Run
python eval.py
. - The efficiency evaluation is supported by the ptflops toolkit. To evaluate the FLOPs and Parameters, run
python eval_complexity.py
.