From df540c2d4f9f7827e21a0b5635717b8fffe5557e Mon Sep 17 00:00:00 2001 From: yayoimizuha Date: Thu, 8 Feb 2024 03:20:06 +0900 Subject: [PATCH] update --- test_script/retinaface_poc.py | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 test_script/retinaface_poc.py diff --git a/test_script/retinaface_poc.py b/test_script/retinaface_poc.py new file mode 100644 index 0000000..8c9fbc2 --- /dev/null +++ b/test_script/retinaface_poc.py @@ -0,0 +1,8 @@ +from retinaface.pre_trained_models import get_model +from PIL import Image +from numpy import array + +image = Image.open("./test_script/manaka_test.jpg") +model = get_model("resnet50_2020-07-20", max_size=2048) +model.eval() +print(model.predict_jsons(array(image)))