登录 注册
当前位置:主页 > 资源下载 > 37 > 使用Keras实现的最小YOLO V1版本:humble-yolo

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

  • 更新:2024-12-16 20:16:38
  • 大小:5KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

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