search

fh2019ustc / PolySnake

The official code for “Recurrent Generic Contour-based Instance Segmentation with Progressive Learning”, TCSVT, 2024.
MIT License
67 stars 7 forks source link

训练Cityspaces时，代码中没有调用cfg.det_model的地方 #13

Open ZitengXue opened 6 months ago

ZitengXue commented 6 months ago

老哥您好！不好意思再次打扰您，我在训练Cityspaces首先预训练了一个检测器 python train_net.py --cfg_file configs/city_snake.yaml model rcnn_det 接着我尝试一起训练polysnake python train_net.py --cfg_file configs/city_snake.yaml model rcnn_snake det_model rcnn_det 但是我发现config中的det_model没有被调用，预训练的检测器权重没有被加载 我发现在snake的代码中有加载检测器权重的代码但是PolySnake中却没有，我可以把这两行代码加上一起训练Cityspaces吗？

ZitengXue commented 6 months ago

还有请问老哥我是否可以在deep snake上训练好检测器的权重直接在PolySnake上加载？