New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
有关self.dct_h和self.dct_w的设置? #10
Comments
@XFR1998 Line 19 in aa5fb63
Line 29 in aa5fb63
|
那如果我换成别的网络不用resnet我应该怎么根据实际情况设置呢? |
我看你好像是根据通道数进行设置的对吗,应该怎么修改呢?谢谢 |
那我觉得你就干脆全部设成7就行了。 |
好的谢谢大佬指点~ |
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
在这个类中MultiSpectralAttentionLayer有以下部分。
if h != self.dct_h or w != self.dct_w:
x_pooled = torch.nn.functional.adaptive_avg_pool2d(x, (self.dct_h, self.dct_w))
# If you have concerns about one-line-change, don't worry. :)
# In the ImageNet models, this line will never be triggered.
# This is for compatibility in instance segmentation and object detection.
如果我的任务是目标检测,我该怎么设置self.dct_h和self.dct_w?
The text was updated successfully, but these errors were encountered: