-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Open
Description
Env
- Nano,
- Ubuntu16.04
- Cuda version 10.04
- TensorRT version 8.1
About this repo
retinaface 和 retinafaceAntiCov
Your problem
retinaface 模型我是在这个路径下载:https://github.com/wang-xinyu/Pytorch_Retinaface

转wts报错:
cuda device count: 1
Traceback (most recent call last):
File "genwts_resnet.py", line 36, in
main()
File "genwts_resnet.py", line 12, in main
net = net.to(device)
AttributeError: 'collections.OrderedDict' object has no attribute 'to'
retinaface_anticov模型下载路径:https://github.com/deepinsight/insightface/tree/master/detection/retinaface_anticov
有一个问题:
按照unzip解压后和gen_wts.py中要求的./cov2/mnet_cov2不匹配了。

Metadata
Metadata
Assignees
Labels
No labels