Prometheus带方向目标分割

Prometheus带方向目标分割

航拍图像中的目标检测是城市规划、交通监控、港口管理和海上救援等众多应用的重要步骤。 航拍图像是从鸟瞰图拍摄的。在航拍图像中检测物体是一项具有挑战性的任务，因为物体通常具有不同的尺度和纹理，而且背景很复杂。此外，目标通常密集地排列并以任意方向显示。因此，将水平边界框应用于定向目标检测将导致检测到的边界框与目标之间的错位。为了解决这个问题，建议使用带方向的目标分割框。

1.1 下载代码、安装依赖

# 注意：首先确认是否安装好了pytorch、cuda
cd <path-to-Prometheus>
./Scripts/install_detection_oriented_object_segs.sh

或者单步执行

# 注意：首先确认是否安装好了pytorch、cuda
cd <path-to-Prometheus>
git clone https://gitee.com/jario-jin/prometheus-oriented-object-segs.git Modules/oriented_detection_segs
cd Modules/oriented_detection_segs
# 安装依赖
sudo apt-get install wget
sudo apt-get install swig
pip3 install shapely -i https://pypi.tuna.tsinghua.edu.cn/simple
# 编译DOTA_devkit模块
cd datasets/DOTA_devkit
swig -c++ -python polyiou.i
python3 setup.py build_ext --inplace
# 回到<path-to-Modules/oriented_detection_segs>
cd ..
cd ..
mkdir weights_dota
# 下载预训练模型
wget http://jario.ren/models/prometheus-oriented-object-segs/model_50.pth -O weights_dota/model_50.pth

1.2 运行代码

cd <path-to-Prometheus>
# Python3运行pytorch代码
./Scripts/start_oriented_object_segs_server.sh
# Python2运行ros接口
roslaunch prometheus_detection oriented_object_segs.launch

输出话题/prometheus/object_detection/oriented_object_segs，参数见launch文件：Modules/object_detection/launch/oriented_object_segs.launch