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

相关推荐
JoinApper6 小时前
目标检测系列-Yolov5下载及运行
人工智能·yolo·目标检测
XiaoMu_0011 天前
基于Django+Vue3+YOLO的智能气象检测系统
python·yolo·django
程序员柳1 天前
基于YOLOv8的车辆轨迹识别与目标检测研究分析软件源代码+详细文档
人工智能·yolo·目标检测
小胖墩有点瘦1 天前
【基于yolo和web的垃圾分类系统】
人工智能·python·yolo·flask·毕业设计·课程设计·垃圾分类
格林威2 天前
棱镜的技术加持:线扫相机如何同时拍RGB和SWIR?
人工智能·深度学习·数码相机·yolo·计算机视觉
大学生毕业题目2 天前
毕业项目推荐:83-基于yolov8/yolov5/yolo11的农作物杂草检测识别系统(Python+卷积神经网络)
人工智能·python·yolo·目标检测·cnn·pyqt·杂草识别
阿崽meitoufa2 天前
[水果目标检测5]AppleYOLO:基于深度OC-SORT的改进YOLOv8苹果产量估计方法
yolo
weixin_377634842 天前
【目标检测】特征理解与标注技巧
yolo·目标检测
笑脸惹桃花2 天前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
weixin_377634842 天前
【YOLO】数据增强bug
yolo·bug