search

zju3dv / snake

Code for "Deep Snake for Real-Time Instance Segmentation" CVPR 2020 oral
Other
1.15k stars 229 forks source link

修改loss函数 #240

Closed xiannsa closed 1 year ago

xiannsa commented 1 year ago

大佬你好，请问要想修改loss函数，应该在哪个文件修改？ 我想使用Siou，这个可以用在snake上面吗？

pengsida commented 1 year ago

需要看你使用的config文件调用了哪个trainer file，一般是下面两个中的一个：

  1. https://github.com/zju3dv/snake/blob/master/lib/train/trainers/snake.py
  2. https://github.com/zju3dv/snake/blob/master/lib/train/trainers/rcnn_snake.py