一般是在终端输入指令运行
实现本地右键运行需要加一串这个代码
class Args:
def __init__(self):
self.model = "path\your\model"
self.image = "path\your\image"
指定即将运行的硬件类型,支持的值为`[cpu, gpu]`
self.device = 'cpu'
#部署模型时使用的后端, 支持的值为`[paddle,pptrt,pplite,ort,openvino,trt]` |openvino|
self.backend = "paddle"
self.topk = 1
args = Args()
如果这个不行的话需要先定义这个类,然后把你配置模型文件的代码封装到主函数里面,最后写上运行的模块
class Args:
def __init__(self):
self.model = "path/your/model"
self.image = "path/your/image"
self.device = "cpu"
self.backend = "paddle"
def main():
args = Args() # 使用 Args 类中的参数
# 配置runtime,加载模型(你的模型)
# 预测结果可视化
if __name__ == "__main__":
main()