JDet
JDet copied to clipboard
RotationInvariantPooling(256, 8) 硬编码?
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?