
humble-yolo:使用Keras的最小YOLO V1实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
humble-yolo是我实现的YOLO v1的一个简洁版本,旨在深入理解该算法卓越的性能。要测试它,请按照以下步骤操作:首先运行generate-dataset.py以创建训练数据集;接着,在main.py中启用--train命令,并设置训练周期数为100步;完成后,您应能观察到带有边框的图像列表,其中前10张图像是未参与训练的数据用于测试。您可以利用这些图像评估模型性能。此外,经过训练后,main.py可以被用来保存当前状态以备后续使用。如果无需进行额外的预训练即可直接运行模型并仅关注结果,请随时调用main.py并将最后权重加载到网络中即可实现功能。
全部评论 (0)
还没有任何评论哟~


