devmaxxing
commented
1 year ago 你在用这个视频吗：https://github.com/oliverfei/videocr-PaddleOCR/blob/master/example_cropped.mp4
如果在用那个视频你的crop_x,crop_y,crop_height,crop_width有问题。在README加了一个例子：
如果视频已经crop过了不用crop就用use_fullframe=True
`from videocr import save_subtitles_to_file
@title OCR parameters
input_file_path = "example_cropped.mp4" #@param {type:"string"} output_file_path = "example.srt" #@param {type:"string"} language_code = "ch" #@param {type:"string"} use_gpu = False #@param {type:"boolean"} start_time = "00:00" #@param {type:"string"} end_time = "" #@param {type:"string"} confidence_threshold = 50 #@param {type:"number"} similarity_threshold = 80 #@param {type:"number"} frames_to_skip = 0 #@param {type:"integer"} crop_x = 40 #@param {type:"integer"} crop_y = 650 #@param {type:"integer"} crop_width = 890 #@param {type:"integer"} crop_height = 69 #@param {type:"integer"}
save_subtitles_to_file(input_file_path, output_file_path, lang=language_code, time_start=start_time, time_end=end_time, conf_threshold=confidence_threshold, sim_threshold=similarity_threshold, use_gpu=use_gpu,
Models different from the default mobile models can be downloaded here: https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.3/doc/doc_en/models_list_en.md
报错： `--------------------------------------------------------------------------- error Traceback (most recent call last) /tmp/ipykernel_8898/783819912.py in
25 # use_fullframe=True, # note: videocr just assumes horizontal lines of text. vertical text scenario hasn't been implemented yet
26 frames_to_skip=frames_to_skip, # can skip inference for some frames to speed up the process
---> 27 crop_x=crop_x, crop_y=crop_y, crop_width=crop_width, crop_height=crop_height)
~/work/y00580163/videocr-PaddleOCR/videocr/api.py in save_subtitles_to_file(video_path, file_path, lang, time_start, time_end, conf_threshold, sim_threshold, use_fullframe, det_model_dir, rec_model_dir, use_gpu, brightness_threshold, similar_image_threshold, similar_pixel_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height) 23 f.write(get_subtitles( 24 video_path, lang, time_start, time_end, conf_threshold, ---> 25 sim_threshold, use_fullframe, det_model_dir, rec_model_dir, use_gpu, brightness_threshold, similar_image_threshold, similar_pixel_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height))
~/work/y00580163/videocr-PaddleOCR/videocr/api.py in get_subtitles(video_path, lang, time_start, time_end, conf_threshold, sim_threshold, use_fullframe, det_model_dir, rec_model_dir, use_gpu, brightness_threshold, similar_image_threshold, similar_pixel_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height) 10 11 v = Video(video_path, det_model_dir, rec_model_dir) ---> 12 v.run_ocr(use_gpu, lang, time_start, time_end, conf_threshold, use_fullframe, brightness_threshold, similar_image_threshold, similar_pixel_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height) 13 return v.get_subtitles(sim_threshold) 14
~/work/y00580163/videocr-PaddleOCR/videocr/video.py in run_ocr(self, use_gpu, lang, time_start, time_end, conf_threshold, use_fullframe, brightness_threshold, similar_image_threshold, similar_pixel_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height) 74 75 if similar_image_threshold: ---> 76 grey = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) 77 if prevgrey is not None: 78 , absdiff = cv2.threshold(cv2.absdiff(prev_grey, grey), similar_pixel_threshold, 255, cv2.THRESH_BINARY)
error: OpenCV(4.6.0) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'`