JDet icon indicating copy to clipboard operation
JDet copied to clipboard
verified publisher icon Jittor

RotationInvariantPooling(256, 8) 硬编码？

Open Xunius opened this issue 3 years ago • 0 comments

Hi,

我学习 s2anet 实现的时候发现在 models/roi_heads/s2anet_head.py 里，S2ANetHead 定义中有 (https://github.com/Jittor/JDet/blob/2fd9ef0833f2edc79ee8a30aadca503ce5054a36/python/jdet/models/roi_heads/s2anet_head.py#L160)：

self.or_pool = RotationInvariantPooling(256, 8)

我猜这里的 256 是不是硬编码了，应该换成 self.feat_channels?

Xunius avatar Aug 18 '22 14:08 Xunius