Unable to load model
#1
by
aidea-dev
- opened
Can someone provide a working script to load this model?
I've tried it all.
Hey @aidea-dev ,
you can download the model and use it in the full e2e iris recognition pipeline using our open source open-iris project (https://github.com/worldcoin/open-iris/tree/main).
In the codebase of open-iris, you can also find examples how to download just model weights from hugging face repo:
- ONNX model download example (https://github.com/worldcoin/open-iris/blob/main/src/iris/nodes/segmentation/onnx_multilabel_segmentation.py#L33)
- TensorRT model download example (https://github.com/worldcoin/open-iris/blob/main/src/iris/nodes/segmentation/tensorrt_multilabel_segmentation.py#L70)
Hope that will help you!
Best regards,
Wiktor