tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.
Apache License 2.0
3.75k stars 470 forks source link

关于CWD_Loss的问题 #121

Open JinYu1998 opened 1 year ago

JinYu1998 commented 1 year ago

Before Asking

Search before asking

Question

这块的代码

y_sy_tshape都是 (N, C, H, W).

 for idx, (s, t) in enumerate(zip(y_s, y_t)):
          assert s.shape == t.shape
          N, C, H, W = s.shape

通过 enumerate 拿到的 stshape 应该是 (C, H, W). 这里的 N是多少呢?

Additional

No response