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

相关推荐
王哈哈^_^3 小时前
【数据集】【YOLO】【目标检测】农作物病害数据集 11498 张,病害检测,YOLOv8农作物病虫害识别系统实战训推教程。
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·1024程序员节
少林and叔叔11 小时前
基于yolov5.7.0的人工智能算法的下载、开发环境搭建(pycharm)与运行测试
人工智能·pytorch·python·yolo·目标检测·pycharm
王哈哈^_^12 小时前
【数据集】【YOLO】【目标检测】建筑垃圾数据集 4256 张,YOLO建筑垃圾识别算法实战训推教程。
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·数据集
B站计算机毕业设计之家13 小时前
计算机视觉:YOLO实现目标识别+目标跟踪技术 pyqt界面 OpenCV 计算机视觉 深度学习 计算机(建议收藏)✅
python·opencv·yolo·计算机视觉·目标跟踪·口罩识别
tt55555555555513 小时前
YOLOv5模型架构详解(三)
yolo
lxmyzzs14 小时前
【图像算法 - 31】基于深度学习的太阳能板缺陷检测系统:YOLOv12 + UI界面 + 数据集实现
人工智能·深度学习·算法·yolo·缺陷检测
lxmyzzs14 小时前
【图像算法 - 32】基于深度学习的风力发电设备缺陷检测系统:YOLOv12 + UI界面 + 数据集实现
深度学习·算法·yolo·计算机视觉
B站计算机毕业设计之家14 小时前
深度学习:YOLOv8人体行为动作识别检测系统 行为识别检测识系统 act-dataset数据集 pyqt5 机器学习✅
人工智能·python·深度学习·qt·yolo·机器学习·计算机视觉
深度学习lover15 小时前
<数据集>yolo煤矿安全帽识别数据集<目标检测>
人工智能·python·深度学习·yolo·目标检测·计算机视觉·煤矿安全帽识别
创思通信1 天前
树莓派的YOLO智能AI识别系统,识别ESP32还是STM32
人工智能·stm32·yolo