项目1 yolov5鱼苗检测计数

yolov5鱼苗检测

  • [1. yolov5鱼苗检测](#1. yolov5鱼苗检测)
    • [1.1. 环境配置](#1.1. 环境配置)
    • [1.2 Predict](#1.2 Predict)
    • [1.3 Validate](#1.3 Validate)
    • [1.4 Train](#1.4 Train)
    • [1.5 生成 ONNX](#1.5 生成 ONNX)
  • [2 代码解析](#2 代码解析)
    • [2.1 模型](#2.1 模型)
    • [2.2 数据集](#2.2 数据集)
    • [2.3 损失函数](#2.3 损失函数)
    • [2.4 训练](#2.4 训练)
    • [2.5 预测](#2.5 预测)

之前做的项目,再回顾一下
环境:GPU1卡,CPU4核,每显卡12GB,内存12GB

1. yolov5鱼苗检测

1.1. 环境配置

Setup

python 复制代码
!git clone https://github.com/ultralytics/yolov5  # clone
%cd yolov5
%pip install -qr requirements.txt  # install

import torch
import utils
display = utils.notebook_init()  # checks

1.2 Predict

python 复制代码
!python classify/predict.py --weights yolov5s-cls.pt --img 224 --source data/images
# display.Image(filename='runs/predict-cls/exp/zidane.jpg', width=600)

1.3 Validate

python 复制代码
# Download Imagenet val (6.3G, 50000 images)
!bash data/scripts/get_imagenet.sh --val

1.4 Train

python 复制代码
# Train YOLOv5s Classification on Imagenette160 for 3 epochs
!python classify/train.py --model yolov5s-cls.pt --data image

1.5 生成 ONNX

python 复制代码
!python export.py --data data/custom_data.yaml --weights weight/best.pt

测试结果

2 代码解析

2.1 模型

2.2 数据集

2.3 损失函数

2.4 训练

2.5 预测

相关推荐
羊羊小栈2 天前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业
大霸王龙4 天前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘
m_136874 天前
Mac M 系列芯片 YOLOv8 部署教程(CPU/Metal 后端一键安装)
yolo·macos
格林威4 天前
机器视觉在半导体制造中有哪些检测应用
人工智能·数码相机·yolo·计算机视觉·视觉检测·制造·相机
、、、、南山小雨、、、、5 天前
YOLO在ubuntu22安装
yolo
羊羊小栈5 天前
基于「YOLO目标检测 + 多模态AI分析」的铁路轨道缺陷检测安全系统(vue+flask+数据集+模型训练)
人工智能·yolo·目标检测·语言模型·毕业设计·创业创新·大作业
Python图像识别5 天前
63_基于深度学习的草莓病害检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
范男5 天前
YOLO11目标检测运行推理简约GUI界面
图像处理·人工智能·yolo·计算机视觉·视觉检测
model20055 天前
ubuntu24.04+5070ti训练yolo模型(2)
人工智能·yolo
强盛小灵通专卖员6 天前
RK3576边缘计算设备部署YOLOv11
人工智能·深度学习·yolo·边缘计算·sci·rk3576·小论文