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)))