用keras/tf实现的语义分割的全神经网络。
模型在models.py中，包括ResNet/DenseNet。
个人有待验证。
依赖:
CUDA, TensorFlow
在Ubuntu 14.04/16.04系统上已经测试了。
所用数据集Pascal VOC，Berkeley，MS COCO
安装：
cd ~/src
# install dependencies
pip install pillow keras sacred
# fork of keras-contrib necessary for DenseNet based models
git clone [email protected]:ahundt/keras-contrib.git -b densenet-atrous
cd keras-contrib
sudo python setup.py install
# Install python coco tools
cd ~/src
git clone https://github.com/pdollar/coco.git
cd coco
sudo python setup.py install
训练测试：
cd ~/src/Keras-FCN
cd utils
# Generate pretrained weights
python transfer_FCN.py
cd ~/src/Keras-FCN
# Run training
python train.py
# Evaluate the performance of the network
python evaluate.py
训练的模型文件在目录~/src/Keras-FCN/Models下。
閱讀更多 AI踐行者 的文章
