You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
279 B
16 lines
279 B
3 years ago
|
import torch
|
||
|
from torchsummary import summary
|
||
|
import numpy as np
|
||
|
|
||
|
from nets import Model
|
||
|
|
||
|
model = Model(max_disp=256, mixed_precision=False, test_mode=True)
|
||
|
model.eval()
|
||
|
|
||
|
t1 = torch.rand(1, 3, 480, 640)
|
||
|
t2 = torch.rand(1, 3, 480, 640)
|
||
|
|
||
|
output = model(t1,t2)
|
||
|
print(output.shape)
|
||
|
|