Есть код который находит мусор на фото, надо его переделать чтобы он определял с видео у меня не получается

Question

Есть код который находит мусор на фото, надо его переделать чтобы он определял с видео у меня не получается

Pipi Pukpuk Ученик (196), открыт 1 неделю назад

вот код с определением мусора

  import yolov5  
 
  
 
# load model  
 
model = yolov5.load('keremberke/yolov5n-garbage')  
 
  
 
# set model parameters  
 
model.conf = 0.25 # NMS confidence threshold  
 
model.iou = 0.45 # NMS IoU threshold  
 
model.agnostic = False # NMS class-agnostic  
 
model.multi_label = False # NMS multiple labels per box  
 
model.max_det = 1000 # maximum number of detections per image  
 
  
 
# set image  
 
img = 'C:/Users/pipip/PycharmProjects/PythonProject/results/u53967t1.png'  
 
  
 
# perform inference  
 
results = model(img, size=640)  
 
  
 
# inference with test time augmentation  
 
results = model(img, augment=True)  
 
  
 
# parse results  
 
predictions = results.pred[0]  
 
boxes = predictions[:, :4] # x1, y1, x2, y2  
 
scores = predictions[:, 4]  
 
categories = predictions[:, 5]  
 
  
 
# show detection bounding boxes on image  
 
results.show()  
 
  
 
# save results into "results/" folder  
 
results.save(save_dir='results/')

а вот видео с вебки

 import cv2  
 
  
 
cap = cv2.VideoCapture(0)  
 
while cap.isOpened():  
 
ret, img = cap.read()  
 
cv2.imshow('output', img)  
 
  
 
key = cv2.waitKey(10)  
 
if key == 27:  
 
break  
 
  
 
cap.release()  
 
cv2.destroyAllWindows()

Answer 1

На

 import yolov5 
import cv2 
 
model = yolov5.load('keremberke/yolov5n-garbage') 
model.conf = 0.25 
model.iou = 0.45 
model.agnostic = False 
model.multi_label = False 
model.max_det = 1000 
 
cap = cv2.VideoCapture(0) 
 
if not cap.isOpened(): 
    print("Не удалось открыть веб-камеру.") 
    exit() 
 
while True: 
    ret, frame = cap.read() 
    if not ret: 
        print("Не удалось считать кадр с веб-камеры.") 
        break 
 
    results = model(frame, size=640) 
    results.render() 
 
    cv2.imshow('Обнаружение мусора', frame) 
 
    if cv2.waitKey(1) & 0xFF == 27: 
        break 
 
cap.release() 
cv2.destroyAllWindows()