-
Notifications
You must be signed in to change notification settings - Fork 20
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
请问关于模型的"img_size"这个参数是什么呀?不需要和输入一样吗? #2
Comments
分块的size。 |
对于长宽不一样的输入图片 这个img_size要设成不一样的吗 |
都可以的~ |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
我看到使用的实例中:
`from SETR.transformer_seg import SETRModel
import torch
if name == "main":
net = SETRModel(img_size=(32, 32),
in_channels=3,
out_channels=1,
hidden_size=1024,
num_hidden_layers=8,
num_attention_heads=16,
decode_features=[512, 256, 128, 64])
t1 = torch.rand(1, 3, 256, 256)
print("input: " + str(t1.shape))
SETRModel(img_size=(32, 32), 有这个,但是下面的输入使用的img_size是(256,256)。所以这个模型的img_size参数对应的是什么呀
The text was updated successfully, but these errors were encountered: