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不匹配了。
Env
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不匹配了。