Permalink
Branch: master
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
15 lines (12 sloc) 512 Bytes
from scannertools import pose_detection, vis, sample_video
import scannerpy
import os
def main():
with sample_video() as video:
db = scannerpy.Database()
frames = [0, 100, 200]
poses = pose_detection.detect_poses(db, videos=[video], frames=[frames])
vis.draw_poses(db, videos=[video], poses=poses, frames=[frames], path='sample_poses.mp4')
print('Wrote video with poses drawn to {}'.format(os.path.abspath('sample_poses.mp4')))
if __name__ == "__main__":
main()