Deep Snake复现中遇到的问题

论文题目:Deep Snake for Real-Time Instance Segmentation

源码链接:https://github.com/zju3dv/snake/


我在复现deep snake这篇文章的代码,作者的readme文档里步骤写的很详细(GitHub - zju3dv/snake: Code for "Deep Snake for Real-Time Instance Segmentation" CVPR 2020 oral),但我在一步一步按照其操作的过程中遇到了一些问题。


主要问题是,按照官方文档里所说的步骤训练和测试数据集时,只有sbd能显示出测试后结果,但在其他数据集上都不显示。

按照这篇说的方法来训练coco数据集Real-Time实例分割DeepSnake使用篇,也没能显示出结果,不知道是哪里出了问题,求助~

环境已经配置完毕,可以运行成功关于sbd的训练、测试和可视化,但是训练其他数据集就不行。

利用SBD数据集运行demo会出现一个可爱的狗子:

#在coco训练执行命令:
python -W ignore train_net.py --cfg_file ./configs/coco_snake.yaml
#按照上述博客改了脚本之后,执行测试coco命令:
python -W ignore run.py --type visualize --cfg_file configs/coco_snake.yaml ct_score 0.3

运行结果及报错内容 :
Q1:在服务器上训练,总是没多久就自动断开连接,训练了很久还是在epoch1,是数据量太大了需要改小batch_size或是其他参数吗?还是其他的问题?

Deep Snake复现中遇到的问题_第1张图片

Q2:测试后并不显示分割轮廓后的结果
执行coco测试后可视化效果图:没有分割轮廓显示

Deep Snake复现中遇到的问题_第2张图片

 

按照官方文档中的Testing on Cityscapes、Testing on Kitti也没有结果显示。

#Testing on Cityscapes
# use coco evaluator
python run.py --type evaluate --cfg_file configs/city_rcnn_snake.yaml
# use the cityscapes official evaluator
python run.py --type evaluate --cfg_file configs/city_rcnn_snake.yaml test.dataset CityscapesVal
#Testing on Kitti
python run.py --type evaluate --cfg_file configs/kins_snake.yaml test.dataset KinsVal


测试没有结果,这是为什么呢?
Deep Snake复现中遇到的问题_第3张图片

可视化图片中也没有任何分割轮廓出来
Deep Snake复现中遇到的问题_第4张图片

正确cityscape数据集测试结果是这样的,但是我训练的时候完全没有出来

Deep Snake复现中遇到的问题_第5张图片

我只能对SBD进行训练和测试,不知道是哪一块没有弄好,还是说训练不同的数据集需要改tran_net里的代码呢?

最近准备认真复现并学习这篇论文,所以尝试写blog看有没有人能交流一下,希望能快点解决这些问题,然后可以更新解决方法!
 

你可能感兴趣的:(deep,snake,pytorch,python,deep,learning,计算机视觉)