架构解析

使用Python实现,相关源码位于py目录下

├── data
├── detector.py
├── lib
    ├── batch_detect.py
    ├── data
        ├── parse_location.py
        └── pascal_voc_07.py
        ├── __init__.py
    ├── models
    │   ├── basic_conv2d.py
    │   ├── __init__.py
    │   ├── location_dataset.py
    │   ├── multi_part_loss.py
    │   ├── __pycache__
    │   └── yolo_v1.py
    ├── __pycache__
    ├── train.py
    └── utils
        ├── draw.py
        ├── file.py
        ├── __init__.py
        ├── __pycache__
        ├── util.py
        └── voc_map.py
└── models
  • 数据集操作
    • lib/data/parse_location.py
    • lib/data/pascal_voc_07.py
  • CNN模型:
    • lib/models/basic_conv2d.py
    • lib/models/yolo_v1.py
  • 损失函数:
    • lib/models/multi_part_loss.py
  • 模型训练:lib/train.py
  • mAP测试:lib/batch_detect.py
  • 检测器实现:detector.py