PaulChongPeng / YOLO2TensorFlow

YOLOv2 implemented by TensorFlow
Apache License 2.0
38 stars 18 forks source link

YOLO2TensorFlow

YOLOv2 implemented by TensorFlow

TODO:

  1. 数据增强
  2. 损失函数优化
  3. 多GPU部署训练
  4. 性能评估脚本
  5. detector.py
  6. preprocess_bboxes 优化
  7. draw box

References:

darknet darkflow YAD2K YOLOv1 paper YOLOv2 paper