run.py has some detail to modify

Question

run.py has some detail to modify

l2009312042 opened this issue 6 years ago · 0 comments

l2009312042 commented 6 years ago

1、TOFlow(intPreprocessedHeight, intPreprocessedWidth,'interp', cuda_flag=CUDA)
need to add interp
2、at function Estimate ，should add the following respect
inTensor = torch.cat((tensorPreprocessedFirst.unsqueeze(0),tensorPreprocessedSecond.unsqueeze(0)), 1)
tensorFlow = torch.nn.functional.interpolate(
input=net(inTensor),
size=(intHeight, intWidth), mode='bilinear', align_corners=False)

and i test some pics ，especially at the edge ，the artifact is obvious ，this flow based method is not well enough