Skip to content

请问retinaface 和 retinafaceAntiCov 原始模型在哪里下载? #1683

@AI-LLM2

Description

@AI-LLM2

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
Image
转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

Image

有一个问题:
按照unzip解压后和gen_wts.py中要求的./cov2/mnet_cov2不匹配了。

Image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions