参考资料 ultralytics/models/yolo/classify/predict.py
备注
该文件可在https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/ yolo/classify/predict .py。如果您发现问题,请通过提交 Pull Request🛠️ 帮助修复。谢谢🙏!
ultralytics.models.yolo.classify.predict.ClassificationPredictor
扩展了 BasePredictor 类,用于基于分类模型进行预测。
说明
- Torchvision 分类模型也可以传递给 "模型 "参数,例如 model='resnet18'。
示例
源代码 ultralytics/models/yolo/classify/predict.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
初始化 ClassificationPredictor,将任务设置为 "分类"。
源代码 ultralytics/models/yolo/classify/predict.py
postprocess(preds, img, orig_imgs)
对预测进行后处理,返回结果对象。
源代码 ultralytics/models/yolo/classify/predict.py
preprocess(img)
将输入图像转换为与模型兼容的数据类型。