Re-open IP-camera videostream if disconnected

[EgOrlukha]

Streaming from IP-cameras via rtsp was freezing when signal lost. I added an option to re-open cv2.Videocapture in case of video-stream disconnection.

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Enhanced error handling for unresponsive video streams in YOLOv5.

📊 Key Changes

• The update function in utils/datasets.py now takes an additional stream parameter.
• Improved error reporting with a warning message if a video stream is unresponsive.
• Added functionality to attempt to re-open the video stream automatically if the connection is lost.

🎯 Purpose & Impact

• Helps users to debug issues with IP camera connections by printing a warning when a video feed is lost.
• Increases robustness by trying to reconnect to the video stream, potentially reducing interruptions during stream processing.
• Facilitates continuity in video frame capture for YOLOv5, which is crucial for applications that rely on consistent real-time video analysis. 💪📹🔄

[github-actions]

👋 Hello @EgOrlukha, thank you for submitting a 🚀 PR! To allow your work to be integrated as seamlessly as possible, we advise you to:

• ✅ Verify your PR is up-to-date with origin/master. If your PR is behind origin/master an automatic GitHub actions rebase may be attempted by including the /rebase command in a comment body, or by running the following code, replacing 'feature' with the name of your local branch:

git remote add upstream https://github.com/ultralytics/yolov5.git
git fetch upstream
git checkout feature  # <----- replace 'feature' with local branch name
git merge upstream/master
git push -u origin -f

• ✅ Verify all Continuous Integration (CI) checks are passing.
• ✅ Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." -Bruce Lee

[glenn-jocher]

@EgOrlukha this seems really useful, thanks for the PR!

Have you been able to see some streams reconnect when they drop due to the PR code?

[glenn-jocher]

@EgOrlukha PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐

[EgOrlukha]

@EgOrlukha this seems really useful, thanks for the PR!

Have you been able to see some streams reconnect when they drop due to the PR code?

@glenn-jocher Yes, I changed the code for my needs and it works stable even if a videostream is lost for some reason. These are OpenCV issues, I just re-open cv2.Videocapture() in case it was disconnected.

[glenn-jocher]

@EgOrlukha great, good to know!