yolov8、win10环境配置

一、准备工作

1. 准备工作: 安装CUDA

首先确定电脑显卡匹配的cuda版本
win+r打开命令,cmd进入命令窗口,输入

nvidia-smi

CUDA匹配版本

在nvidia官网(链接: nvidia官网.)下载对应的cuda版本,我用的是离线下载,防止中途网络问题导致安装不成功。我选择的是12.2.2版本
cuda官网版本
下载好之后,一路默认安装。
安装完成后,还是在命令窗口,输入

nvcc --version

cuda版本号
我安装了两个版本的cuda,可以查到两个cuda版本号。
在系统环境变量中查看一下是否有环境变量
cuda环境变量1
cuda环境变量path

2. 准备工作:在GitHub上下载最新的项目文件

GitHub项目链接: github项目地址.
GitHub下载项目文件

二、配置环境

1. 解压项目文件,右键在pycharm中打开(我安装的是最新的pycharm)

GitHub项目地址使用方法
在这里插入图片描述
在这里插入图片描述
安装指令
以下两条指令用来安装环境

pip install ultralytics # 安装
pip install -U ultralytics # 也可以用这个,安装并更新
#按照官方文档的说法这行语句会安装所有需要的环境,但是我输入这个语句之后,还是需要安装其他安装包,直接pip install 就可以了

2. 构建数据集

2.1 划分数据集

我是将10000张自己的数据集抽出一些,分60%作为训练集,20%作为验证集,%20作为测试集。
labelme打标签,但是yolo的标签格式不同,所以需要将标签格式转换,脚本代码如下:
标签文件-格式转换脚本文件.

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import copy
from lxml.etree import Element, SubElement, tostring, ElementTree

import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join

classes = ["AT1", "AF1",
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值