doublekuk

>  我是修改的这里，不使用self.is_vis，否则会返回shape[0]，就会遇到问题。 我修改了这个红框里的`return pred_occ.shape[0]`为`return pred_occ`，然后模型可以正常预测了