Skip to content

zhengshuchen/HCFNet

Repository files navigation

HCF-Net

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:

Model name IoU nIoU
UCF Net 80.09 78.31

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages