search

Jittor / JDet

JDet is an object detection benchmark based on Jittor. Mainly focus on aerial image object detection (oriented object detection).
Apache License 2.0
191 stars 34 forks source link

RotationInvariantPooling(256, 8) 硬编码？ #52

Open Xunius opened 2 years ago

Xunius commented 2 years ago

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?