ReDet icon indicating copy to clipboard operation
ReDet copied to clipboard

Published 20 hours ago • verified publisher icon csuhan

关于configs中的num_classes以及能否实例分割问题

Open 1994su opened this issue 4 years ago • 1 comments

您好，我想请教一下，比如我想利用redet训练5个类别，我单纯改config中的numclass可以吗？我现在直接这样修改之后会报错误，还是需要修改一下checkpoint中的类别数目？ redet设置文件中如果加入mask_head和mask_roi_extractor能对目标进行实例分割吗？必须用实例分割的标签才可以吧？

1994su avatar Sep 04 '21 09:09 1994su

  1. 需要改config和对应dataset中的CLASSES
  2. 目前代码中去掉了mask相关内容，可以参考cascade rcnn，及基于RoITrans的实例分割https://ieeexplore.ieee.org/abstract/document/9323274

csuhan avatar Sep 06 '21 03:09 csuhan