pytorch 在YOLOv5中隐藏标签和配置值

9wbgstp7  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(317)

我加载了一个模型(在我的笔记本电脑中)。我试图使用(model.hide_labels = True)和(model.hide_conf = True)从图像中删除(隐藏)标签和配置值。但没有成功。
我如何隐藏标签和配置值?我只想要盒子。

model = torch.hub.load('ultralytics/yolov5', 'custom', './yolov5s/best.pt')  # custom trained model

## Images

im = './Data/val/images/10_8.jpg'  # or file, Path, URL, PIL, OpenCV, numpy, list

model.hide_labels = True
model.hide_conf = True

## Inference

results = model(im)

## Results

results.show()  # or .show(), .save(), .crop(), .pandas(), print()
u4dcyp6a

u4dcyp6a1#

我通过使用results.show(标签=错误)解决了该问题

相关问题