如何用GPU加速检测

AIZOOTech / FaceMaskDetection

开源人脸口罩检测模型和数据 Detect faces and determine whether people are wearing mask.

MIT License

2.08k stars 618 forks source link

Open yilawu opened 4 years ago

yilawu commented 4 years ago

想用GPU加速检测，现在的速度是0.15s一帧

yilawu commented 4 years ago

I got it

Ndron commented 4 years ago

model.cuda() model.eval()