我有没有办法在视频中间裁剪以在那部分运行我的车辆检测模型

Question

我想仅在视频中心推断我的车辆检测模型。就像您在此

Answer 1

这是我用于作物的代码，有点有效。我只需要使用我的车辆检测模型来实现它，我将在完成后立即发布

# Import packages
import cv2
import numpy as np

# Open the video
cap = cv2.VideoCapture('test.mp4')

# Initialize frame counter
cnt = 0

# Some characteristics from the original video
w_frame, h_frame = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), 
int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps, frames = cap.get(cv2.CAP_PROP_FPS), cap.get(cv2.CAP_PROP_FRAME_COUNT)

# Here you can define your croping values
x,y,h,w = 0,0,600,1000

# output
fourcc = cv2.VideoWriter_fourcc(*'MP4V')

#fourcc = cv2.cv.CV_FOURCC(*'XVID')

out = cv2.VideoWriter('result.mp4', fourcc, fps, (w, h))


# Now we start
while(cap.isOpened()):
ret, frame = cap.read()

    cnt += 1 # Counting frames

# Avoid problems when video finish
if ret==True:
    # Croping the frame
    crop_frame = frame[y:y+h, x:x+w]

    # Percentage
    xx = cnt *100/frames
    print(int(xx),'%')

    # Saving from the desired frames
    #if 15 <= cnt <= 90:
    #    out.write(crop_frame)

    # I see the answer now. Here you save all the video
    out.write(crop_frame)

    # Just to see the video in real time          
    cv2.imshow('frame',frame)
    cv2.imshow('croped',crop_frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
else:
    break


cap.release()
out.release()  
cv2.destroyAllWindows()