Add depth to the input channel of YOLO

[PedroPiana]

Hello, I'm a student, and I'm trying to add a depth channel as an input to YOLO in order to try to improve it. How can I do this?

[UltralyticsAssistant]

👋 Hello @PedroPiana, thank you for reaching out to the Ultralytics community 🚀! Your project using Kinect to extract image measurements sounds fascinating. While we can't answer your question directly here, we recommend you look into our Docs for guidance on YOLO customization.

If this is a 🐛 Bug Report, please ensure to provide a minimum reproducible example so that we can understand the issue better.

For your custom training ❓ question about adding a depth channel, please provide as much detail as possible, including any existing code snippets you've attempted, and check out our Tips for Best Training Results.

Don't forget to join the vibrant Ultralytics community where you can engage with other users or talk to us directly. Here are some options for you:

• Chat with us live on Discord 🎧.
• Participate in discussions on Discourse.
• Share insights or find help on our Subreddit.

Upgrade

Ensure you are using the latest version of the ultralytics package along with all requirements in a Python>=3.8 environment with PyTorch>=1.8. You can do this by running:

pip install -U ultralytics

Verified Environments

Here are some environments where YOLO can be run:

• Notebooks with free GPU:
• Google Cloud Deep Learning VM: Follow the GCP Quickstart Guide
• Amazon Deep Learning AMI: Check the AWS Quickstart Guide
• Docker Image: Use the Docker Quickstart Guide

Status

If this badge is green, all tests are passing for YOLO Modes and Tasks. An Ultralytics engineer will also assist you shortly.

[glenn-jocher]

@PedroPiana to add a depth channel to YOLO, you'll need to modify the model architecture to accept additional input channels and adjust the data preprocessing pipeline accordingly. For detailed guidance, you might want to explore the Ultralytics documentation or community forums.