FazBrowse GitHub Viewer | Trending | | Home

Tools: [Download Repo ZIP] [Original HTTPS Page]

DeepLearning/Image_segmentation/DeepLabV3 at master · Overmind7/DeepLearning · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
config		config
data/test		data/test
dataLoader		dataLoader
models		models
utils		utils
README.md		README.md
deeplabv3_resnet50.png		deeplabv3_resnet50.png
display_dataset.py		display_dataset.py
palette.json		palette.json
predict.py		predict.py
train.py		train.py

README.md

DeepLabV3 网络结构复现

论文题目：《Rethinking Atrous Convolution for Semantic Image Segmentation》
论文地址：https://arxiv.org/abs/1706.05587

训练数据下载

链接：https://pan.baidu.com/s/1sRomCSZ-Wo_qlCMMQMbkIA?pwd=9noc 提取码：9noc

预训练权重

训练

配置文件：config/example.yaml

您需要在配置文件中修改您的数据路径，详细注解见config/example.yaml

train

python train.py --config_name config/example.yaml

推理

python predict.py --cfg config/example.yaml --weights [checkpoint_path] --source [test_data_path]

tensorboard

tensorboard --logdir=log_path

Back | FazBrowse Home | New Git URL