吴� � 2021-12-20 12:49 采纳率: 80%
浏览 1410
已结题

请问YOLO已经训练好了自己的模型文件,best.pt的文件,如何写代码调用这个文件,输出检测的数量和准确度等信息。

请问YOLO已经训练好了自己的模型文件,.pt的文件,如何写代码调用这个文件,输出检测的数量和准确度等信息。
以下代码不对吗?错误在哪里 现在就是同意的用这个 模型文件 在测试的 yolov5中的 detect.py中运行的结果很好,但是我自己写的下面的代码就不太行,显示出来结果差~


#import torch
#import numpy as np
#import cv2
#model = torch.hub.load('ultralytics/yolov5', 'custom', path='C:\\Users\\Administrator\\Desktop\\ceshi1\\best.pt')
#frame = cv2.imread('C:\\Users\\Administrator\\Desktop\\11\\0.jpg')
#frame = cv2.resize(frame,(640,640))
#results = model(frame)
#cv2.imshow('yolo', np.squeeze(results.render()))
#print(results)
#cv2.waitKey(0)
#

```cv2.destroyAllWindows()
  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2021-12-20 13:33
    关注

    model的输入一般是是4维的【bs,c,h,w】,你这么写只有三个维度,而且也没有转成tensor,能运行就是奇迹了,没有报维度不匹配?

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 12月28日
  • 已采纳回答 12月20日
  • 创建了问题 12月20日

悬赏问题

  • ¥15 stm32代码移植没反应
  • ¥15 matlab基于pde算法图像修复,为什么只能对示例图像有效
  • ¥100 连续两帧图像高速减法
  • ¥15 组策略中的计算机配置策略无法下发
  • ¥15 如何绘制动力学系统的相图
  • ¥15 对接wps接口实现获取元数据
  • ¥20 给自己本科IT专业毕业的妹m找个实习工作
  • ¥15 用友U8:向一个无法连接的网络尝试了一个套接字操作,如何解决?
  • ¥30 我的代码按理说完成了模型的搭建、训练、验证测试等工作(标签-网络|关键词-变化检测)
  • ¥50 mac mini外接显示器 画质字体模糊