python用YOLOv8对图片进行分类

用yolov8的模型进行分类

  1. 先上效果图

  2. 图片资源

  3. 模型下载地址

    https://github.com/ultralytics/ultralytics

  4. 代码

py 复制代码
import matplotlib.pyplot as plt
from ultralytics import YOLO
from PIL import Image
import cv2

model = YOLO('../ultralytics/yolov8n.pt')

# print(model.names)

img_path = '../ultralytics/ultralytics/assets/bus.jpg'

img = cv2.imread(img_path)
results = model(img_path)

preds = results[0].boxes.xyxy.cpu().numpy().astype('uint32')

colors = plt.cm.get_cmap('hsv', len(model.model.names))

for index, pred in enumerate(preds):
    i = int(results[0].boxes.cls[index].item())
    color = colors(i)
    color = (color[0] * 200, color[1] * 255, color[2] * 200)
    img = cv2.rectangle(img, (int(pred[0]), int(pred[1])), (int(pred[2]), int(pred[3])), color, 2)
    img = cv2.putText(img, results[0].names[i], (int(pred[0]), int(pred[1])), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2)

cv2.imshow("1", img)
cv2.waitKey()
cv2.destroyAllWindows()

'''
print('..........')
print(len(results[0].names))                        # 所有的分类名
print(len(results[0].boxes.cls))                    # 结果分类
print(results[0].boxes.conf)                        # 置信度
print(results[0].boxes.cls)                         # 分类类别
print(type(int(results[0].boxes.cls[0].item())))    # 类别索引
print(results[0].names[0])                          # 分类的名称(可以根据类别索引进行获取)
print(results[0].boxes.xyxy)                        # 左上角和右下角坐标
'''
相关推荐
nbsaas-boot1 小时前
Java 正则表达式白皮书:语法详解、工程实践与常用表达式库
开发语言·python·mysql
仗剑_走天涯1 小时前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
chao_7896 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
烛阴6 小时前
Python装饰器解除:如何让被装饰的函数重获自由?
前端·python
noravinsc6 小时前
django 一个表中包括id和parentid,如何通过parentid找到全部父爷id
python·django·sqlite
ajassi20007 小时前
开源 python 应用 开发(三)python语法介绍
linux·python·开源·自动化
沉默媛7 小时前
如何安装python以及jupyter notebook
开发语言·python·jupyter
Deng9452013148 小时前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
2401_878624798 小时前
pytorch 自动微分
人工智能·pytorch·python·机器学习