DG-YOLO icon indicating copy to clipboard operation
DG-YOLO copied to clipboard

domain_label代码不太理解

Open luoluo123123123123 opened this issue 1 year ago • 1 comments

image image 请问,为什么只有7个type,domain_label却要初始化8个0,我的猜想是第0维度是batch_size, 但是我不理解这个mask处理,为什么要把batch_szie这一维==1的挑出来,就是mask这段不太理解在做什么。

luoluo123123123123 avatar Apr 27 '23 07:04 luoluo123123123123

这是一个很生草的处理,因为数据集定义的时候第一个type是type1不是type0。 如果我把type1的一拿出来制作domain_labels,然后只定义7个type,然后你就会发现domain_labels[:,7]报错了QvQ。 我觉得你应该懂我啥意思。

mousecpn avatar Apr 27 '23 07:04 mousecpn