jetson inference怎么设置用自己的权重来检测
时间: 2023-03-28 15:01:08 浏览: 149
您好,关于 jetson inference 怎么设置用自己的权重来检测,您可以按照以下步骤进行操作:
1. 将您的权重文件放置在 jetson-inference/python/training/detection/ssd/models 目录下。
2. 在 jetson-inference/python/training/detection/ssd 目录下,打开 detectnet.py 文件。
3. 在 detectnet.py 文件中,找到如下代码:
model = jetson.inference.detectNet(opt.network, sys.argv, opt.threshold)
4. 将其修改为:
model = jetson.inference.detectNet(opt.network, sys.argv, opt.threshold)
model.SetModelPath("models/your_weight_file_name.onnx")
model.LoadModel()
5. 将 "your_weight_file_name.onnx" 替换为您的权重文件名。
6. 保存并关闭 detectnet.py 文件。
7. 在终端中进入 jetson-inference/python/training/detection/ssd 目录下,运行以下命令:
python3 detectnet.py --model=your_model_name
其中,"your_model_name" 为您的模型名称。
希望以上步骤能够帮助您解决问题。
阅读全文