MODEL_ZOO.md

Model Zoo and Baselines

Introduction

This guide presents our Model Zoo, designed for construction site applications, featuring models trained on construction site datasets. These models are optimized for single GPU use in supervised training.

Pre-trained Models and Fine-tuned Variants

We offer original and fine-tuned models for construction site monitoring tasks like object detection and segmentation. Models are available for download via Google Drive.

DETR (Detection Transformer)

• Pre-trained DETR (ResNet-50 Backbone): A DETR model with a ResNet-50 backbone, ready for custom fine-tuning.

Object Detection Fine-tuned DETR

• Object Detection Fine-tuned DETR: A DETR model fine-tuned for construction site object detection.

Segmentation Fine-tuned DETR

• Segmentation Fine-tuned DETR: Adapted for segmentation in construction sites, enhancing analysis capabilities.

Original Mask2Former

• Original Mask2Former: A leading model for segmentation without the need for fine-tuning.

Fine-tuned Mask2Former

• Fine-tuned Mask2Former: Optimized for construction site imagery, enhancing performance and accuracy.