PIXIE icon indicating copy to clipboard operation
PIXIE copied to clipboard

Demo on video files is too slow

Open HYLee1008 opened this issue 4 years ago • 1 comments

Running demo_fit_body.py on video file is too slow. RCNN is running on cpu mode now. You need to change this line

https://github.com/YadiraF/PIXIE/blob/3162310d1507221aebca46721a0ec92d35d98cc4/demos/demo_fit_body.py#L31

to this

    testdata = TestData(args.inputpath, iscrop=args.iscrop, body_detector='rcnn', device=device)

HYLee1008 avatar Nov 16 '21 04:11 HYLee1008

+1. Getting a ~2.25x speedup on a P40 GPU compared to CPU.

hshreeshail avatar May 18 '22 09:05 hshreeshail