猪分割算法代码使用

Project info

将猪和盘分割出来

Training

1、下载vgg16.npy，并将其放在文件的根目录下; 2、运行train.py用于开始训练；

Inference

inference的方式有两种，test.py和test_with_pb.py分别用于对ckpt类型的模型和对pb类型的模型 进行测试。