从视频帧中裁剪检测到 object

Question

我有一个 model 可以检测并在手周围绘制一个边界框，还有一个 function 可以检测到 object。 当我传递之前从视频中提取的单个帧时，它可以工作，但是这种方法的问题是我得到的每个框的大小略有不同，而我需要所有框具有相同的宽度和高度。

然后我尝试调用我的 function 来裁剪代码中的边界框，以便从视频中提取帧。 我工作得很好，我从视频中提取的每一帧都得到了相同大小的边界框，但是一旦到达第二个视频，我就会收到错误：

文件“C:\Users\Public\handdetection.py”，第 49 行，在 draw_box_on_image1 中返回 image_np[p1[1]:p2[1], p1[0]:p2[0]]

TypeError：'NoneType' object 不可下标

这是提取的代码

def draw_box_on_image1(num_hands_detect, score_thresh, scores, boxes, im_width, im_height, image_np):
for i in range(num_hands_detect):
    if (scores[i] > score_thresh):
        (left, right, top, bottom) = (boxes[i][1] * im_width, boxes[i][3] * im_width,
                                      boxes[i][0] * im_height, boxes[i][2] * im_height)
        p1 = (int(left-100), int(top-100))
        p2 = (int(right+100), int(bottom+100))
        return cv2.rectangle(image_np, p1,p2, (77, 255, 9), 3, 1)
        return image_np[p1[1]:p2[1], p1[0]:p2[0]]


for i in tqdm(range(train.shape[0])):
count = 0
videoFile = train['video_name'][i] # i is each row (each video) from column video name in dataframe train


cap = cv2.VideoCapture(videoFile)# capturing the video from the given path
frameRate = cap.get(cv2.CAP_PROP_FPS) #frame rate
x=1
while(cap.isOpened()):
    frameId = cap.get(10) #current frame number
    ret, frame = cap.read()
    image_np=frame
    image_np=draw_box_on_image1(num_hands_detect, args.score_thresh,scores, boxes, im_width, im_height,
                             image_np)
    if (ret != True):
        break
    if (frameId % math.floor(frameRate) == 0):
        # storing the frames in a new folder named train_1
        #filename ='train_1/' + videoFile.split('/')[1].split(' ')[0] +"_frame%d.jpg" % count;count+=1
        filename ='train_1/' + videoFile +"_frame%d.jpg" % count;count+=1
        cv2.imwrite(filename, image_np)
cap.release()

你们能帮我解决这个问题吗？