-
使用Keras实现的最小YOLO V1版本:humble-yolo
资源介绍
humble-yolo是我写的YOLO v1的最小实现,目的是了解惊人的YOLO算法。
要测试它,请运行:
generate-dataset.py生成数据
main.py --train --epoch 100用于训练网络
您应该看到带有边框的图像列表。 前10张图像是未用于训练的测试数据。 您可以在这些网络上评估网络的性能。 其余图像已用于训练。
main.py可以在完成训练后节省重量。 如果您想在不经过培训的情况下运行网络并仅查看结果,则仅运行main.py即可加载最后的权重并重新显示结果。
- 上一篇: yolo源码+tensorflow版本
- 下一篇: keras-yolo3-master.rar