mousecpn / DG-YOLO

[ICIP2020] TOWARDS DOMAIN GENERALIZATION IN UNDERWATER OBJECT DETECTION
47 stars 5 forks source link

domain_label代码不太理解 #12

Open luoluo123123123123 opened 1 year ago

luoluo123123123123 commented 1 year ago

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

mousecpn commented 1 year ago

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