项目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 预测

相关推荐
AI军哥19 分钟前
MySQL8的安装方法
人工智能·mysql·yolo·机器学习·deepseek
白熊1887 小时前
【计算机视觉】CV实战项目 - 基于YOLOv5的人脸检测与关键点定位系统深度解析
人工智能·yolo·计算机视觉
一勺汤11 小时前
YOLOv11改进-双Backbone架构:利用双backbone提高yolo11目标检测的精度
人工智能·yolo·双backbone·double backbone·yolo11 backbone·yolo 双backbone
豆芽81915 小时前
图解YOLO(You Only Look Once)目标检测(v1-v5)
人工智能·深度学习·学习·yolo·目标检测·计算机视觉
Eric.Lee202115 小时前
数据集-目标检测系列- F35 战斗机 检测数据集 F35 plane >> DataBall
人工智能·算法·yolo·目标检测·计算机视觉
白熊1881 天前
【计算机视觉】CV实战项目 - 基于YOLOv5与DeepSORT的智能交通监控系统:原理、实战与优化
人工智能·yolo·计算机视觉
FPGA开源工坊1 天前
FPGA上实现YOLOv5的一般过程
yolo·fpga开发
埃菲尔铁塔_CV算法2 天前
YOLO 模型的深度剖析及其在生物医药领域的创新应用
深度学习·神经网络·yolo·目标检测·计算机视觉
limengshi1383922 天前
使用Python将YOLO的XML标注文件转换为TXT文件格式
xml·python·yolo
unix2linux3 天前
YOLO v5 Series - FFmpeg & (HTML5 + FLV.js ) & ONNX YOLOv5s Integrating
yolo·ffmpeg·html5