error: (-215:Assertion failed) Yolov5 Readnet Error - opencv

Question

from PyQt5 import QtGui, QtWidgets, uic
from PyQt5.QtGui import QPixmap
import sys
import cv2 #opencv-python==4.5.3
import time
from PyQt5.QtCore import pyqtSignal, pyqtSlot, Qt, QThread
from PyQt5.QtWidgets import QApplication, QWidget, QListWidget, QLabel, QVBoxLayout
from PyQt5.QtWidgets import *
from PyQt5.QtWidgets import (QApplication, QMainWindow, QPushButton, QPlainTextEdit,
QVBoxLayout, QWidget)
from PyQt5.QtCore import QProcess
import numpy as np
import sqlite3
from numpy.core.records import array
import torch
def dict_factory(cursor, row):
d = {}
for idx, col in enumerate(cursor.description):
d[col[0]] = row[idx]
return d
stop_threads = False
class VideoThread(QThread):
CONFIDENCE_THRESHOLD = 0.2
NMS_THRESHOLD = 0.4
COLORS = [(0, 255, 255), (255, 255, 0), (0, 255, 0), (255, 0, 0)]
frame_counter=0
class_names = []
with open("backup/coco.names",'r') as f:
class_names = [cname.strip() for cname in f.readlines()]
#net = torch.hub.load('C:/Users/Niko/Desktop/Versuch/backup/custom_yolov5s.yaml', 'custom', 'C:/Users/Niko/Desktop/Versuch/backup/best_Neu.pt', force_reload=true)
net = cv2.dnn.readNetFromONNX(r'C:/Users/Niko/Desktop/Versuch/backup/best.onnx') #r'C:/Users/Niko/Desktop/Versuch/backup/yolov5s.onnx',
#net = cv2.dnn.readNet("C:/Users/Niko/Desktop/Versuch/backup/yolov4-tiny-custom_NEU.cfg", r"C:/Users/Niko/Desktop/Versuch/backup/best_Neu.pt", 'custom', source = 'local', path='last' ) #sonst (backup/yolo.weights)"
try:
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)
print("Using GPU")
except print(0):
print("Using CPU")
model = cv2.dnn_DetectionModel(net)
#model.setInputParams(size=(640,640), scale=1/255, swapRB=True) #416,416
model.setBlob(size=(640,640), scale=1/255, swapRB=True) #416,416
change_pixmap_signal = pyqtSignal(np.ndarray)
#ekran_goruntusu_ates_signal = pyqtSignal(np.ndarray)
screenshot_fire_signal = pyqtSignal(np.ndarray)
#ekran_goruntusu_duman_signal = pyqtSignal(np.ndarray)
screenshot_smoke_signal = pyqtSignal(np.ndarray)
#konsol_signal=pyqtSignal(str)
konsolen_signal=pyqtSignal(str)
#nesne_ismi_signal = pyqtSignal(str,int)
objektname_signal = pyqtSignal(str,int)
def run(self):
global stop_threads
vc = cv2.VideoCapture("drohnenvideo_feuer1080.mp4")
while cv2.waitKey(1) < 1:
(grabbed, frame) = vc.read()
if grabbed:
frame=self.window_in(frame)
self.change_pixmap_signal.emit(frame)
if stop_threads:
break
vc.release()
I want to build an custom object detection and therefore trained a Yolov5 model on a custom data set. and recieved the best weights and converted them into .onnx. This Code is working for yolov4 but for yolov5 i converted it to .onnx but unfortunately it stills brings me an error
cv2.error: OpenCV(4.5.3) C:\Users\runneradmin\AppData\Local\Temp\pip-req-build-sn_xpupm\opencv\modules\dnn\src\onnx\onnx_importer.cpp:2146: error: (-2:Unspecified error) in function 'cv::dnn::dnn4_v20210608::ONNXImporter::handleNode'
Node [Identity]:(onnx::Reshape_582) parse error: OpenCV(4.5.3) C:\Users\runneradmin\AppData\Local\Temp\pip-req-build-sn_xpupm\opencv\modules\dnn\src\dnn.cpp:5298: error: (-215:Assertion failed) inputs.size() in function 'cv::dnn::dnn4_v20210608::Layer::getMemoryShapes'