架构解析¶
使用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