Intel
/
dpt-hybrid-midas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Younes Belkada 2022-12-07 12:50:02 +00:00 committed by huggingface-web
parent 25883a7ba6
commit 402bc4afca
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -64,6 +64,7 @@ prediction = torch.nn.functional.interpolate(
mode="bicubic", mode="bicubic",
align_corners=False, align_corners=False,
) )
```
# visualize the prediction # visualize the prediction
output = prediction.squeeze().cpu().numpy() output = prediction.squeeze().cpu().numpy()