search

ying09 / TextFuseNet

A PyTorch implementation of "TextFuseNet: Scene Text Detection with Richer Fused Features".
MIT License
476 stars 123 forks source link

代码现在是否只支持batch_size==1? #34

Closed jtcjump closed 3 years ago

jtcjump commented 3 years ago

Mutil_Path_Fuse_Module：：forward if self.training: proposal_boxes = proposals[0].proposal_boxes classes = proposals[0].gt_classes else: proposal_boxes = proposals[0].pred_boxes classes = proposals[0].pred_classes

    if len(proposal_boxes) == 0:
          return x

代码中只取了proposals[0]，batch_size >1时 text = x[char_pos[i]] 会报错

Real-YeJ commented 3 years ago

@jtcjump 是的，后续会更新。