DeepLabv3Plus-Pytorch-ROS

このパッケージは、DeepLabV3Plus-Pytorchに基づくパクってつくたROSパケージです。

インストール

このリポジトリをcatkin_wsのsrcにクローンして

$ git clone https://github.com/Tsumoridesu/deeplabv3_plus_pytorch_ros.git

ビルド

$ cd ../
$ catkin build deeplabv3_plus_pytorch_ros
$ source ~/catkin_ws/devel/setup.bash 

Quick Start

1. 必要パケージのインストール

$ pip install -r requirements.txt

2. 起動

$ roslaunch deeplabv3_plus_pytorch_ros segmentation.launch

3. パラメータ調整

dataset:

VOCデータセットとCityscapes２種類があります

VOCの場合：year 2012_aug

Cityscapesの場合 : cityscapes

model:

DeepLabV3	DeepLabV3+
deeplabv3_resnet50	deeplabv3plus_resnet50
deeplabv3_resnet101	deeplabv3plus_resnet101
deeplabv3_mobilenet	deeplabv3plus_mobilenet
deeplabv3_hrnetv2_48	deeplabv3plus_hrnetv2_48
deeplabv3_hrnetv2_32	deeplabv3plus_hrnetv2_32

好きのモデルを選択すればいい

ckpt：

Pascal VOC2012 Aug：

Model	Batch Size	FLOPs	train/val OS	mIoU	Dropbox	Tencent Weiyun
DeepLabV3-MobileNet	16	6.0G	16/16	0.701	Download	Download
DeepLabV3-ResNet50	16	51.4G	16/16	0.769	Download	Download
DeepLabV3-ResNet101	16	72.1G	16/16	0.773	Download	Download
DeepLabV3Plus-MobileNet	16	17.0G	16/16	0.711	Download	Download
DeepLabV3Plus-ResNet50	16	62.7G	16/16	0.772	Download	Download
DeepLabV3Plus-ResNet101	16	83.4G	16/16	0.783	Download	Download

Cityscapes：

Model	Batch Size	FLOPs	train/val OS	mIoU	Dropbox	Tencent Weiyun
DeepLabV3Plus-MobileNet	16	135G	16/16	0.721	Download	Download
DeepLabV3Plus-ResNet101	16	N/A	16/16	0.762	Download	Comming Soon

好きのウェイトをダンロードもしくはカスタマイズのウェイトをcheak_pointsに入れる

ほかのパラメータ：

本家に参照して適当に変える

ほぼ同じだから多分大丈夫

ノードと話題なら適当に変えばいい、多分

Note

Issuesは中国語と日本語対応、英語じゃなくでもいい

Ubuntu 20.04　Python3.9で動けるはず、動けない場合なら多分自力で解決するほうが早い