Yuvraj SeegolamandClaude Sonnet 4.5
9ec12e4c17
- Implement ColoredLogger with lightning emoji for enhanced visual feedback - Add color-coded log levels (INFO=Green, WARNING=Yellow, ERROR=Red) - Refactor model configurations to config/models.json for easier customization - Integrate huggingface_hub for reliable downloads with progress tracking - Add tqdm progress bar styled to match logger format for batch processing - Improve ONNX model version detection for DA3, DAD, and custom models - Reorganize codebase: move nodes to nodes/ directory, utilities to separate files - Update dependencies: add huggingface_hub to requirements.txt - Enhance error handling and logging throughout engine builder and depth processing Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
73 lines
3.5 KiB
JSON
73 lines
3.5 KiB
JSON
{
|
|
"v1_small": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-Onnx/resolve/main/depth_anything_vits14.onnx",
|
|
"filename": "depth_anything_vits14.onnx",
|
|
"engine_name": "v1_depth_anything_vits14-fp16.engine"
|
|
},
|
|
"v1_base": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-Onnx/resolve/main/depth_anything_vitb14.onnx",
|
|
"filename": "depth_anything_vitb14.onnx",
|
|
"engine_name": "v1_depth_anything_vitb14-fp16.engine"
|
|
},
|
|
"v1_large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-Onnx/resolve/main/depth_anything_vitl14.onnx",
|
|
"filename": "depth_anything_vitl14.onnx",
|
|
"engine_name": "v1_depth_anything_vitl14-fp16.engine"
|
|
},
|
|
"v2_small": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-2-Onnx/resolve/main/depth_anything_v2_vits.onnx",
|
|
"filename": "depth_anything_v2_vits.onnx",
|
|
"engine_name": "v2_depth_anything_vits-fp16.engine"
|
|
},
|
|
"v2_base": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-2-Onnx/resolve/main/depth_anything_v2_vitb.onnx",
|
|
"filename": "depth_anything_v2_vitb.onnx",
|
|
"engine_name": "v2_depth_anything_vitb-fp16.engine"
|
|
},
|
|
"v2_large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-2-Onnx/resolve/main/depth_anything_v2_vitl.onnx",
|
|
"filename": "depth_anything_v2_vitl.onnx",
|
|
"engine_name": "v2_depth_anything_vitl-fp16.engine"
|
|
},
|
|
"v2_metric_hypersim_large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-2-Onnx/resolve/main/depth_anything_v2_metric_hypersim_vitl.onnx",
|
|
"filename": "depth_anything_v2_metric_hypersim_vitl.onnx",
|
|
"engine_name": "v2_depth_anything_metric_hypersim_vitl-fp16.engine"
|
|
},
|
|
"v2_metric_vkitti_large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-2-Onnx/resolve/main/depth_anything_v2_metric_vkitti_vitl.onnx",
|
|
"filename": "depth_anything_v2_metric_vkitti_vitl.onnx",
|
|
"engine_name": "v2_depth_anything_metric_vkitti_vitl-fp16.engine"
|
|
},
|
|
"DAD_small": {
|
|
"url": "https://huggingface.co/yuvraj108c/distill-any-depth-onnx/resolve/main/Distill-Any-Depth-Multi-Teacher-Small.onnx",
|
|
"filename": "Distill-Any-Depth-Multi-Teacher-Small.onnx",
|
|
"engine_name": "Distill-Any-Depth-Multi-Teacher-Small-fp16.engine"
|
|
},
|
|
"DAD_base": {
|
|
"url": "https://huggingface.co/yuvraj108c/distill-any-depth-onnx/resolve/main/Distill-Any-Depth-Multi-Teacher-Base.onnx",
|
|
"filename": "Distill-Any-Depth-Multi-Teacher-Base.onnx",
|
|
"engine_name": "Distill-Any-Depth-Multi-Teacher-Base-fp16.engine"
|
|
},
|
|
"DAD_large": {
|
|
"url": "https://huggingface.co/yuvraj108c/distill-any-depth-onnx/resolve/main/Distill-Any-Depth-Multi-Teacher-Large.onnx",
|
|
"filename": "Distill-Any-Depth-Multi-Teacher-Large.onnx",
|
|
"engine_name": "Distill-Any-Depth-Multi-Teacher-Large-fp16.engine"
|
|
},
|
|
"DAD_large_2w_iter": {
|
|
"url": "https://huggingface.co/yuvraj108c/distill-any-depth-onnx/resolve/main/Distill-Any-Depth-Dav2-Teacher-Large-2w-iter.onnx",
|
|
"filename": "Distill-Any-Depth-Dav2-Teacher-Large-2w-iter.onnx",
|
|
"engine_name": "Distill-Any-Depth-Dav2-Teacher-Large-2w-iter-fp16.engine"
|
|
},
|
|
"DA3Mono-Large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-3-Onnx/resolve/main/DA3Mono-Large.onnx",
|
|
"filename": "DA3Mono-Large.onnx",
|
|
"engine_name": "DA3Mono-Large-fp16.engine"
|
|
},
|
|
"DA3Metric-Large": {
|
|
"url": "https://huggingface.co/yuvraj108c/Depth-Anything-3-Onnx/resolve/main/DA3Metric-Large.onnx",
|
|
"filename": "DA3Metric-Large.onnx",
|
|
"engine_name": "DA3Metric-Large-fp16.engine"
|
|
}
|
|
}
|