Nguyễn Quốc Việt commited on
Upload STRUCTURE_OVERVIEW.md with huggingface_hub
Browse files- STRUCTURE_OVERVIEW.md +49 -0
STRUCTURE_OVERVIEW.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Repository Structure Overview
|
| 2 |
+
|
| 3 |
+
```
|
| 4 |
+
highway-vehicle-detection-code/
|
| 5 |
+
├── models/ # Trained model weights
|
| 6 |
+
│ ├── yolov8m_stage2_improved_best.pt # Final model (recommended)
|
| 7 |
+
│ └── yolov8m_stage1_smart_best.pt # Stage 1 model
|
| 8 |
+
├── training_runs/ # Complete training structure
|
| 9 |
+
│ ├── stage1_smart/
|
| 10 |
+
│ │ ├── args.yaml # Training arguments
|
| 11 |
+
│ │ └── weights/
|
| 12 |
+
│ │ └── last.pt # Last epoch weights
|
| 13 |
+
│ └── stage2_improved/
|
| 14 |
+
│ ├── args.yaml # Training arguments
|
| 15 |
+
│ ├── weights/
|
| 16 |
+
│ │ └── last.pt # Last epoch weights
|
| 17 |
+
│ ├── BoxF1_curve.png # F1 score curve
|
| 18 |
+
│ ├── BoxPR_curve.png # Precision-Recall curve
|
| 19 |
+
│ └── labels.jpg # Label distribution
|
| 20 |
+
├── training_logs/ # Training metrics and visualizations
|
| 21 |
+
│ ├── stage2_results.png # Training results
|
| 22 |
+
│ ├── stage2_confusion_matrix.png # Confusion matrix
|
| 23 |
+
│ ├── stage2_results.csv # Detailed metrics
|
| 24 |
+
│ └── stage2_val_batch0_pred.jpg # Validation samples
|
| 25 |
+
├── finetune_dataset/ # Fine-tuning dataset (92 images)
|
| 26 |
+
│ ├── images/ # Training images
|
| 27 |
+
│ ├── labels/ # Annotation files
|
| 28 |
+
│ ├── README.dataset.txt # Dataset info
|
| 29 |
+
│ ├── README.roboflow.txt # Roboflow info
|
| 30 |
+
│ └── README.md # Dataset overview
|
| 31 |
+
├── dataset_configs/ # Dataset configurations
|
| 32 |
+
│ ├── main_data.yaml # Main dataset config
|
| 33 |
+
│ └── finetune_data.yaml # Fine-tuning config
|
| 34 |
+
├── main.py # Main application
|
| 35 |
+
├── example_usage.py # Usage examples
|
| 36 |
+
├── requirements.txt # Dependencies
|
| 37 |
+
├── test_improved_model.bat # Testing script
|
| 38 |
+
├── PROJECT_REPORT.md # Complete documentation
|
| 39 |
+
├── TEAM_INSTRUCTIONS.md # Team setup guide
|
| 40 |
+
└── README.md # This file
|
| 41 |
+
```
|
| 42 |
+
|
| 43 |
+
## Key Files
|
| 44 |
+
|
| 45 |
+
- **models/**: Ready-to-use trained models
|
| 46 |
+
- **training_runs/**: Complete training structure (like local runs/ folder)
|
| 47 |
+
- **finetune_dataset/**: The 92-image fine-tuning dataset
|
| 48 |
+
- **main.py**: Complete vehicle detection application
|
| 49 |
+
- **External links**: YouTube demo and Kaggle dataset
|