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