Multi-scale and multi-level enhanced features for ship target recognition in complex environments.

Requirement

python 3.8

Pytorch >=1.7

torchvision >=0.8

Training

1. Download datatsets for MMEF (e.g. MAR-ships, CIB-ships, game-of-ships etc) and organize the structure as follows:

dataset

└── train/test

    ├── class_001
    
    |      ├── 1.jpg    
    |      ├── 2.jp
    |      └── ...    
    ├── class_002
    
    |      ├── 1.jpg
    |      ├── 2.jpg
    |      └── ...
    └── ...

2、Train from scratch with train.py.

Citation

Please cite our paper if you use MMEF in your work.

@InProceedings{du2023fine,
  title={Multi-scale and multi-level enhanced features for ship target recognition in complex environments},
  author={Hao Meng; Yang Tian; Fei Yuan},
}

MAR-ships dataset link:

ARGOS-Venice boat classification

website：https://pan.baidu.com/s/1OHBMLMXvkKima1nK5gF-4A?pwd=GLPM 
word：GLPM 

game-of-ships dataset link:

website：https://pan.baidu.com/s/1XkSwtPnxKblxZ6YQV4tEDA?pwd=GLPM 
word：GLPM