13 lines
290 B
Python
13 lines
290 B
Python
|
|
||
|
|
from model.backbone.resnext import ResNeXt50,ResNeXt101,ResNeXt152
|
||
|
|
import torch
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
input=torch.randn(50,3,224,224)
|
||
|
|
resnext50=ResNeXt50(1000)
|
||
|
|
# resnext101=ResNeXt101(1000)
|
||
|
|
# resnext152=ResNeXt152(1000)
|
||
|
|
out=resnext50(input)
|
||
|
|
print(out.shape)
|
||
|
|
|