search

lpiccinelli-eth / UniDepth

Universal Monocular Metric Depth Estimation
Other
462 stars 39 forks source link

I want to run this script on grayscale image #26

Open Abhinav-974 opened 2 months ago

lpiccinelli-eth commented 2 months ago

You can just repeat the single channel and feed that to the infer method. From the demo script, you can change line 11 to rgb_torch = torch.from_numpy(rgb).unsqueeze(0).repeat(3,1,1)