开发者社区 问答 正文

请给出一个使用EasyCV进行推理的示例代码。

请给出一个使用EasyCV进行推理的示例代码。

展开
收起
花开富贵111 2024-08-19 09:42:23 36 分享 版权
1 条回答
写回答
取消 提交回答
  • 使用EasyCV进行推理的示例代码如下:

    import cv2 
    from easycv.predictors.classifier import TorchClassifier 
    
    output_ckpt = 'work_dirs/classification/cifar10/r50/epoch_350_export.pth' 
    tcls = TorchClassifier(output_ckpt) 
    
    img = cv2.imread('aeroplane_s_000004.png') 
    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) 
    output = tcls.predict([img]) 
    print(output)
    
    2024-08-19 15:51:31
    赞同 6 展开评论
问答地址: