search

ppogg / onnx-scrfd-flask

onnx+onnxruntime+scrfd+flask+web
36 stars 14 forks source link

请问scrfd里面提取的特征值在哪。 #1

Open enginefuture opened 3 years ago

enginefuture commented 3 years ago

如题。想做人脸比对，但是看半天没找到检测的特征值是哪个。ty。

ppogg commented 3 years ago

兄弟，你试下打印出向前推理的张量，在/mmdet/apis/inference.py文件里边： 

# We don't restore `torch.is_grad_enabled()` value during concurrent
    # inference since execution can overlap
    torch.set_grad_enabled(False)
    result = await model.aforward_test(rescale=True, **data)
    return result

上述中print（result）

enginefuture commented 3 years ago

感谢了

enginefuture commented 3 years ago

兄弟，你试下打印出向前推理的张量，在/mmdet/apis/inference.py文件里边：

# We don't restore `torch.is_grad_enabled()` value during concurrent
    # inference since execution can overlap
    torch.set_grad_enabled(False)
    result = await model.aforward_test(rescale=True, **data)
    return result

上述中print（result）

还想请教一个问题，现在看到的关键点是5个，我找到了kpss，想请问，如何增加检测的关键点，是不是需要重新训练才行