Skip to content

Gaoshuaikun/MBAN

Repository files navigation

Multi-branch Attention Network (MBAN) for small object detection

Dataset

  1. PASCAL VOC: http://host.robots.ox.ac.uk/pascal/VOC/
  2. NWPU VHR-10: https://github.com/Gaoshuaikun/NWPU-VHR-10

Predict

  1. In the mban.py file, modify model_path and classes_path to correspond to the trained files.
  2. run predict.py.

Evaluate

The evaluation results can be obtained by running get_map.py, and the evaluation results will be saved in the map_out folder.