From da73be0ec299aa2875e92699b770b095943505b0 Mon Sep 17 00:00:00 2001 From: Thomas De Date: Fri, 2 Feb 2024 20:10:09 +0100 Subject: [PATCH] Update README.md --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index 17fdbdf..3e40671 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,10 @@ # face-parsing-onnx face parsing / face masking converted to onnx model +CPU or GPU + +Requirements: opencv-python, numpy, onnxruntime (onnruntime-gpu, cudatoolkit=11.2 cudnn=8.1.0) + python inference.py --source_image "image.jpg" --parser_index 1,2,3,4,5,6,10,11,12,13