一般是在终端输入指令运行

实现本地右键运行需要加一串这个代码

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()

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐