Spaces:
Running on Zero
Running on Zero
Commit History
-
fix: match v5 checkpoint architecture and benchmark scaled inference
Verified signature5878871
-
Add object detection and visualization features
- Implement object extraction from segmentation and height maps in `infer/objects.py`. - Add class map previews and detected objects overlay in `viz/classes.py`. - Update `app.py` to save segmentation and object data as artifacts. - Enhance README with details on object classes and 3D objects. - Create tests for object extraction in `tests/test_objects.py`. - Update requirements and configuration files for new dependencies.b710eca
-
Add DINOv3 model components and loss functions
- Implemented DPT trunk architecture in models/dpt.py for feature extraction. - Added exponential moving average functionality in models/ema.py to stabilize training. - Created DINOv3 encoder in models/encoder.py for robust block finding and layer-wise learning rate decay. - Developed heads for regression and classification tasks in models/heads.py, including gated fusion mechanism. - Introduced custom loss functions in models/losses.py to address specific training challenges. - Implemented test-time augmentation strategies in models/tta.py to enhance model predictions. - Updated requirements.txt to include necessary dependencies for the new model components.40d7c69