DG-YOLO
DG-YOLO copied to clipboard
domain_label代码不太理解
请问,为什么只有7个type,domain_label却要初始化8个0,我的猜想是第0维度是batch_size,
但是我不理解这个mask处理,为什么要把batch_szie这一维==1的挑出来,就是mask这段不太理解在做什么。
这是一个很生草的处理,因为数据集定义的时候第一个type是type1不是type0。 如果我把type1的一拿出来制作domain_labels,然后只定义7个type,然后你就会发现domain_labels[:,7]报错了QvQ。 我觉得你应该懂我啥意思。