关于YOLOX上添加mask rcnn的分支

[yjcreation]

添加mask rcnn的mask分支，针对数据预处理的部分，需要进行修改，修改的时候去调用了环境中default_collate这个部分，此处对数据进行batch数据的整合成[B,C,H,W]。针对数据预处理添加mask，每张图片的实例个数不一样，所以没法直接把tensor整合成[B,C,H,W]的维度。有什么办法解决吗？

[FateScript]

You could patch the mask to the same shape and concat them. #Boxes in different image share the same logic like mask. You may refer these code in YOLOX and apply it to mask.

[grace2015]

想问一下，如何添加这一分支？