我加载了一个模型(在我的笔记本电脑中)。我试图使用(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()
1条答案
按热度按时间u4dcyp6a1#
我通过使用results.show(标签=错误)解决了该问题