code/test_on_video_dockerface.py
@@ -126,6 +126,7 @@ ret,frame = video.read() if ret == False: break cv2_frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB) while True: x_min, y_min, x_max, y_max, conf = int(float(line[1])), int(float(line[2])), int(float(line[3])), int(float(line[4])), float(line[5]) @@ -146,7 +147,7 @@ x_max = min(frame.shape[1], x_max) y_max = min(frame.shape[0], y_max) # Crop image img = frame[y_min:y_max,x_min:x_max] img = cv2_frame[y_min:y_max,x_min:x_max] img = Image.fromarray(img) # Transform
