Skip to content

seakafuka/yolo-fruit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

3bf7d1b · Dec 20, 2024

History

5 Commits
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024
Dec 20, 2024

Repository files navigation

yolo-fruit

项目介绍

本项目基于yolov8n模型实现,前端使用gradio框架,用于水果类型检测

目录结构描述

├── frontend // 包含前端运行文件代码

│ ├── demo.ipynb // 前端运行后,点击网页链接进入前端测试

├── label_data // 本小组图片标注

│ ├── Annotations // 包含标注的xml文件

│ ├── images // 图片

│ ├── imageSets

│ ├── labelmap.txt

├── test // 测试集

├──Txtojson // 包含项目格式转化文件

│ ├── class.txt

└── best.pt //包含训练好的模型参数

└── data.yaml //模型训练参数设置,"""需要更改"""

└── predict.py //预测文件

└── Readme.md //介绍文档

使用说明

一、安装环境依赖

pip install ultralytics
pip install gradio

二、修改data.yaml文件

将"val: E:\vscode_code\apple_and banana_final\test\images"中的 "E:\vscode_code\apple_and banana_final"改为你电脑上的文件地址

三、 运行predict.py文件和frontend文件夹中的demo.ipynb

注:iou指标计算基于第一次valid结果,因此如果第一遍没有跑通,后续再跑需要删除文件夹目录中新产生的runs文件夹,这样才可以得到正确的iou指标。其他指标不受影响

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published