humble-yolo:使用Keras的最小YOLO V1实现

时间:2024-05-20 19:51:43
【文件属性】:

文件名称:humble-yolo:使用Keras的最小YOLO V1实现

文件大小:5KB

文件格式:ZIP

更新时间:2024-05-20 19:51:43

Python

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


【文件预览】:
humble-yolo-master
----requirements.txt(2KB)
----generate-dataset.py(2KB)
----README.md(631B)
----main.py(7KB)

网友评论