HCF-Net

English | 简体中文

HCF-Net is a framework for Infrared Small Object Segmentation

Dataset Structe

If you want to train on custom datasets you should paper dataset as following structure:

|-SIRST
    |-trainval
        |-images
            |-xxx.png
        |-masks
            |-xxx.png
    |-test
        |-images
            |-xxx.png
        |-masks
            |-xxx.png

Training

To train the model, run this command:

python train.py --opt ./options/train.yaml

Evaluation

To evaluate pretrained model, run:

python test.py --opt ./options/test.yaml

Pre-trained Models and Results

You can download pretrained models (we also provide the whole training logs) here:

• HCF for SIRST

Model name	IoU	nIoU
UCF Net	80.09	78.31