yolov5 with matplotlib

[qjl1244281167]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Detection

Bug

When I use detect.py, it seems that I cannot use matplotlib.pyplot as plt drawing in the main function. Although I have imported the relevant code here, I ensure that my environment is correct and the code is correct, but plt cannot Display pictures in pycharm

Environment

No response

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[github-actions]

👋 Hello @qjl1244281167, thank you for your interest in YOLOv5 🚀! Please visit our ⭐️ Tutorials to get started, where you can find quickstart guides for simple tasks like Custom Data Training all the way to advanced concepts like Hyperparameter Evolution.

If this is a 🐛 Bug Report, please provide a minimum reproducible example to help us debug it.

If this is a custom training ❓ Question, please provide as much information as possible, including dataset image examples and training logs, and verify you are following our Tips for Best Training Results.

Requirements

Python>=3.8.0 with all requirements.txt installed including PyTorch>=1.8. To get started:

git clone https://github.com/ultralytics/yolov5  # clone
cd yolov5
pip install -r requirements.txt  # install

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI tests verify correct operation of YOLOv5 training, validation, inference, export and benchmarks on macOS, Windows, and Ubuntu every 24 hours and on every commit.

Introducing YOLOv8 🚀

We're excited to announce the launch of our latest state-of-the-art (SOTA) object detection model for 2023 - YOLOv8 🚀!

Designed to be fast, accurate, and easy to use, YOLOv8 is an ideal choice for a wide range of object detection, image segmentation and image classification tasks. With YOLOv8, you'll be able to quickly and accurately detect objects in real-time, streamline your workflows, and achieve new levels of accuracy in your projects.

Check out our YOLOv8 Docs for details and get started with:

pip install ultralytics

[glenn-jocher]

@qjl1244281167 hi there! 🌟 Matplotlib can sometimes have trouble displaying images in certain environments. This may be due to PyCharm's default settings. You can resolve this issue by adding the following line before importing pyplot:

import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt

This should allow you to display images using plt.show() in PyCharm. For more information, you can also refer to our documentation at https://docs.ultralytics.com/yolov5/. Thank you for your willingness to submit a PR – your contributions are greatly appreciated!

[qjl1244281167]

@qjl1244281167 hi there! 🌟 Matplotlib can sometimes have trouble displaying images in certain environments. This may be due to PyCharm's default settings. You can resolve this issue by adding the following line before importing pyplot:

import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt

This should allow you to display images using plt.show() in PyCharm. For more information, you can also refer to our documentation at https://docs.ultralytics.com/yolov5/. Thank you for your willingness to submit a PR – your contributions are greatly appreciated!
ImportError: Cannot load backend 'TkAgg' which requires the 'tk' interactive framework, as 'headless' is currently running
This is the problem I'm having .I am connected to a remote interpreter on linux

[glenn-jocher]

@qjl1244281167 I see. It seems like the 'TkAgg' backend is not compatible with the headless environment you are running. In this case, you can try using a non-interactive backend such as 'agg' by setting it before importing pyplot:

import matplotlib
matplotlib.use('agg')
import matplotlib.pyplot as plt

This should allow you to display images using plt.show() in a headless environment. Remember to refer to our documentation at https://docs.ultralytics.com/yolov5/ for more information. Thank you for bringing this issue to our attention!

[qjl1244281167]

@qjl1244281167 I see. It seems like the 'TkAgg' backend is not compatible with the headless environment you are running. In this case, you can try using a non-interactive backend such as 'agg' by setting it before importing pyplot:

import matplotlib
matplotlib.use('agg')
import matplotlib.pyplot as plt

This should allow you to display images using plt.show() in a headless environment. Remember to refer to our documentation at https://docs.ultralytics.com/yolov5/ for more information. Thank you for bringing this issue to our attention!

No, this has no effect. In my project, I only need to cancel the module related to yolov5 detect without making any changes, and the image of matplotlib.pyplot can be displayed successfully.

[glenn-jocher]

@qjl1244281167 I understand. It seems there might be some interference between the YOLOv5 detection module and matplotlib. To isolate the issue, you can try removing the YOLOv5 detection module and check if matplotlib works as expected. This might help you identify the source of the problem. Also, make sure that there are no conflicting dependencies or interference between the two. If the issue persists, please feel free to provide more details so we can further assist you. Thank you for your patience and for your willingness to help with the PR!

[qjl1244281167]

@qjl1244281167我明白。看来YOLOv5检测模块和matplotlib之间可能存在一些干扰。要隔离问题，您可以尝试删除 YOLOv5 检测模块并检查 matplotlib 是否按预期工作。这可能会帮助您确定问题的根源。另外，请确保两者之间不存在冲突的依赖关系或干扰。如果问题仍然存在，请随时提供更多详细信息，以便我们进一步为您提供帮助。感谢您的耐心等待以及您愿意帮助公关！

The problem now is that I need to use detect.py to get the xyxy coordinates of the detection object. I must use detect.py, so I extracted this part as a module of my project. I don’t know what to do now. The problem is that as soon as the detected image is processed by yolov5, matplotlib will malfunction. Obviously, there must be an internal conflict between yolov5 and matplotlib.

[glenn-jocher]

@qjl1244281167 I see, I understand your situation better now. It seems that there might be an internal conflict between YOLOv5 and matplotlib when used together. In this case, you can consider encapsulating the parts of YOLOv5's detect.py that you need into a separate function or script, and then try running the matplotlib code separately from that function or script to prevent any interference.

If this doesn't resolve the issue, it might be helpful to reach out to the YOLOv5 community for further insights, as they might have encountered similar challenges and found effective solutions. Additionally, you can also consider raising this issue to the Ultralytics team, who are actively working on YOLOv5, for their guidance.

Thank you for your understanding and for your contribution to the community!

[github-actions]

👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.

For additional resources and information, please see the links below:

• Docs: https://docs.ultralytics.com
• HUB: https://hub.ultralytics.com
• Community: https://community.ultralytics.com

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLO 🚀 and Vision AI ⭐

[qiang92]

Add this line before plt.show() can also solve the problem.
plt.switch_backend('Qt5Agg')
The reason is that once you import DetectMultiBackend module, the matplotlib auto switch backend from QtAgg to agg.

[qjl1244281167]

Add this line before plt.show() can also solve the problem. plt.switch_backend('Qt5Agg') The reason is that once you import DetectMultiBackend module, the matplotlib auto switch backend from QtAgg to agg.

Thanks guy, I will try it.

[glenn-jocher]

Hi @qjl1244281167,

Thank you for sharing your solution! It's great to hear that switching the backend with plt.switch_backend('Qt5Agg') resolves the issue. This insight about the backend switch when importing the DetectMultiBackend module is very valuable.

If you encounter any further issues or have additional questions, please don't hesitate to ask. Also, make sure you're using the latest versions of torch and YOLOv5 from https://github.com/ultralytics/yolov5 to ensure optimal performance and compatibility.

Your contributions and feedback are greatly appreciated by the entire YOLO community and the Ultralytics team! 😊

[nagakokobot]

@qjl1244281167 I ran into similar problem. When I load the yolov5 module for inference, I cannot visualize any image using matplotlib library. Were you able to solve the issue?

edit: Found a workaround in #2779