apple / ml-depth-pro

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.
Other
3.76k stars 253 forks source link

Update run.py #51

Open Yash-2707 opened 1 month ago

Yash-2707 commented 1 month ago

Enhanced error handling and logging Performance improvements Added a --batch-size argument for potential batch processing (not implemented in the inference loop, but prepared for future enhancement). Updated matplotlib figure creation for better layout. and added colorbar to depth map visualization Added file extension check to process only image files.