CaoWGG / CenterNet-CondInst

Instance Segmentation based on CenterNet and CondInst
MIT License
166 stars 27 forks source link

instance_masks的大小每张图不一样,最后如何组batch的勒? #15

Open mmxuan18 opened 3 years ago

mmxuan18 commented 3 years ago

instance_masks = np.zeros((num_objs if num_objs > 0 else 1, output_h, output_w), dtype=np.float32)

这里instance_masks的大小是和objs数相关的,那一个batch中每张图可能不一样,这个在组batch的时候如何处理?