Skip to content

YangBo0411/MTMLNet

Repository files navigation

MTMLNet: Multi-task Mutual Learning Network for Infrared Small Target Detection and Segmentation

The overall architecture

image

Usage

Requirements

Tested on Linux , with Python 3.8, PyTorch 2.1.2, cuda 11.8. pip install the requirements.txt

Dataset

In data\custom.yaml, change the path of the corresponding detection and segmentation label dataset. image

Visualization results

image

Quantitative results

The model of MTMLNet weight can be downloaded from Google Drive or with BaiduYun Drive.

Metheds

R(target)

P(target)

F1(target)

AP

R(pixel)

P(pixel)

F1(pixel)

IoU

MDvsMA

/

/

/

/

0.383

0.435

0.407

 /

ACM

/

/

/

/

0.735

0.806

0.769

0.627

ALC

/

/

/

/

0.540

0.860

0.663

0.639

ISNet

/

/

/

/

0.802

0.695

0.744

0.625

DNANet

/

/

/

/

0.784

0.801

0.793

0.656

UIU

/

/

/

/

0.832

0.214

0.340

0.612

Ours

/

/

/

/

0.819

0.826

0.822

0.692

OSCAR

/

/

/

0.620

/

/

/

/

EFL

0.747

0.775

0.761

0.767

/

/

/

/

MSML

0.773

0.423

0.547

0.752

/

/

/

/

SSTNet

0.355

0.581

0.441

0.659

/

/

/

/

Ours

0.818

0.747

0.781

0.814

/

/

/

/

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages