Varroa_ASRM_training
收藏资源简介:
Varroa 数据集是一个用于训练和评估 YOLOv8-P2 ASRM 变体的对象检测数据集,专注于瓦螨感染图像。该数据集包含感染图像,其中类别 token 3 被映射到类别 1。原始存档保留训练/验证/测试的原始布局,通过 `prepare_dataset.py` 进行去重并创建可复现的 70/15/15 分割布局(使用固定种子 42)。训练超参数包括:图像尺寸 640、批次大小 16、100 个 epoch、早停耐心值 20、工作线程数 4,以及训练种子 42/43。数据集支持三种 ASRM 架构变体:p2_asrm_fusion、input_asrm_auxiliary 和 input_asrm_guided_p2。版本 2 暴露辅助注意力图但无辅助损失,其检测路径与 P2 基线等效。输出包括每个 case/seed 的 `test_metrics.json` 以及比较 CSV 文件(原始运行和聚合统计)。
The Varroa dataset is an object detection dataset designed for training and evaluating YOLOv8-P2 ASRM variants, focusing on Varroa mite infestation images. It contains infected images where class token 3 is mapped to class 1. The original archive retains the original train/val/test layout, and a reproducible 70/15/15 split layout is created with deduplication via `prepare_dataset.py` using a fixed seed 42. Training hyperparameters include image size 640, batch size 16, 100 epochs, early stopping patience 20, number of workers 4, and training seeds 42/43. The dataset supports three ASRM architecture variants: p2_asrm_fusion, input_asrm_auxiliary, and input_asrm_guided_p2. Version 2 exposes auxiliary attention maps without auxiliary loss, and its detection path is equivalent to the P2 baseline. Outputs include `test_metrics.json` per case/seed and comparison CSV files (raw runs and aggregated statistics).
数据集概述
数据集名称:Varroa: three YOLOv8-P2 ASRM variants
数据集链接:https://huggingface.co/datasets/HoangTrungNguyen/Varroa_ASRM_training
数据集内容
该数据集用于训练三种基于 YOLOv8-P2 的 ASRM(Attention-based Spatial Regularization Module) 变体,专门针对 Varroa(瓦螨)检测任务。数据集与 LEVIR-Ship 和原始 Varroa 项目隔离,作为独立训练包提供。
三种 ASRM 架构变体:
p2_asrm_fusioninput_asrm_auxiliaryinput_asrm_guided_p2
数据划分与预处理
- 原始划分:保留原始 train/val/test 布局。
- 标准化划分:通过
prepare_dataset.py脚本对正样本 CSV 记录去重,并生成可复现的 70/15/15 Ultralytics 格式划分。 - 统一转换:所有架构变体和训练种子均使用同一转换后的数据集。
训练配置
训练流程遵循 Varroa/YOLO_custom/HF_YOLO_version_workflow.ipynb 中的规范:
| 配置项 | 值 |
|---|---|
| 训练图像 | 仅感染图像(gt_one) |
| 类别映射 | 类别 token 3 映射到类别 1 |
| 转换种子 | 固定为 42(split seed) |
| 预训练模型 | yolov8n.pt |
| 图像尺寸 | 640 |
| Batch Size | 16 |
| 训练轮数 | 100 |
| 早停耐心值 | 20 |
| 工作线程数 | 4 |
| 训练种子 | 42 / 43 |
输出结果
- 每个案例/种子生成
test_metrics.json文件。 comparison_varroa_asrm.csv包含所有原始运行结果。comparison_varroa_asrm_aggregate.csv包含跨种子的均值/标准差统计。
版本说明(Version 2)
- Version 2 暴露了辅助注意力图,但 没有辅助损失。
- 因此,在本次实验中,其检测路径与 P2 基线故意设计为等效,用于对比分析。




