This commit is contained in:
yayoimizuha 2024-02-08 03:20:06 +09:00
parent 1b6494955d
commit df540c2d4f
1 changed files with 8 additions and 0 deletions

View File

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