CREStereo Repository for the 'Towards accurate and robust depth estimation' project
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.
|
import torch
|
|
|
|
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)
|
|
|
|
|