ARG-NCTU/rtdetr-spscd_buoy_finetuned_e50

RT-DETR Training Results

Model Information

• Model: runs/rtdetr/spscd_base_1280_b8_e100/weights/best.pt
• Dataset: finetune_dataset.yaml
• Classes: 13
• Image Size: 1280×1280
• Batch Size: 8
• Epochs: 50
• Workers: 16

Training Configuration

• Initial Learning Rate: 1e-05
• Final Learning Rate Factor: 0.01
• Patience: 50
• Mixed Precision (AMP): Enabled

Training Time

• Start: 2025-12-24 01:50:05
• End: 2025-12-24 03:34:38
• Duration: 1h 44m 33s (6,273 seconds)

Final Test Results

• mAP50: 0.1216
• mAP50-95: 0.1010

Files in this Repository

• best.pt - Best model checkpoint
• results.csv - Complete training metrics
• results.png - Training curves visualization
• confusion_matrix.png - Confusion matrix
• F1_curve.png, PR_curve.png - Performance curves
• samples/ - Sample validation predictions

Note: TensorBoard logs are included if available. You can also visualize training using results.csv and the curve images.

How to Use

from ultralytics import RTDETR

# Load model
model = RTDETR('best.pt')

# Run inference
results = model('your_image.jpg')

Visualizing Training Progress

• View training curves in results.png
• Analyze per-class performance in confusion matrices
• Check results.csv for detailed epoch-by-epoch metrics