apm1467 / videocr

Extract hardcoded subtitles from videos using machine learning
MIT License
506 stars 117 forks source link

关于设置读帧的stride方法 #7

Open Phincle opened 4 years ago

Phincle commented 4 years ago

你好,非常感谢你,videocr对我的帮助非常大。 但是我现在机器算力有点慢,我想通过增大读帧间隔的方法来提高一下效率,请问怎么修改呢?

lhy1024 commented 4 years ago

https://github.com/apm1467/videocr/blob/9b97c996570897b5a45d1f8b4f046aebcbcca300/videocr/video.py#L46

可以在这里设置步长,比如

frames = (v.read()[1] for _ in range(0,num_ocr_frames,20))
hlthu commented 4 years ago

但是这样做的话,得到的时间戳也是缩放了吧?

lhy1024 commented 4 years ago

但是这样做的话,得到的时间戳也是缩放了吧?

想了一下,确实存在这种情况。因为我的使用场景是提取文本,而不是提取软字幕,所以就没有考虑这方面