ARG-NCTU/rtdetr-spscd_base_1280_b8_e100

RT-DETR Training Results

Model Information

• Model: rtdetr-l.pt
• Dataset: spscd_only_yolo/dataset.yaml
• Classes: 12
• Image Size: 1280×1280
• Batch Size: 8
• Epochs: 100
• Workers: 16

Training Configuration

• Initial Learning Rate: 0.0001
• Final Learning Rate Factor: 0.01
• Patience: 50
• Mixed Precision (AMP): Enabled

Training Time

• Start: 2025-12-22 15:03:32
• End: 2025-12-24 00:27:24
• Duration: 33h 23m 51s (120,231 seconds)

Final Test Results

• mAP50: 0.9919
• mAP50-95: 0.8667

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