Function main

example.py:157–371

Main function with argument parsing and execution.

()

    return depth_colored

def main():
    """Main function with argument parsing and execution."""
    parser = argparse.ArgumentParser(
        description='LingBot-Depth Example: Refine depth maps using the model',
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  # Run with default settings (example 0)
  python example.py

  # Use a different example directory
  python example.py --example 1

  # Use Hugging Face model
  python example.py --model robbyant/lingbot-depth-postrain-dc-vitl14

  # Use local checkpoint
  python example.py --model ckpt/model.pt

  # Custom output directory
  python example.py --output my_results
        """
    )

    parser.add_argument(
        '--example', type=str, default='0',
        help='Example directory number (default: 0)'
    )
    parser.add_argument(
        '--model', type=str,
        default='robbyant/lingbot-depth-pretrain-vitl-14-v0.5',
        help='Model path or Hugging Face ID (default: robbyant/lingbot-depth-pretrain-vitl-14)'
    )
    parser.add_argument(
        '--output', type=str, default='result',
        help='Output directory (default: result)'
    )
    parser.add_argument(
        '--device', type=str, default='auto',
        choices=['auto', 'cuda', 'cpu'],
        help='Device to use (default: auto - uses CUDA if available)'
    )
    parser.add_argument(
        '--no-mask', action='store_true',
        help='Disable masking of invalid regions'
    )

    args = parser.parse_args()

    # Print header
    print("=" * 70)
    print("LingBot-Depth Example: Depth Refinement".center(70))
    print("=" * 70)

    # Setup device
    if args.device == 'auto':
        device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    else: