Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Unable to train model on VisDrone #13054

Closed
1 of 2 tasks
InderSethi opened this issue May 30, 2024 · 7 comments
Closed
1 of 2 tasks

Unable to train model on VisDrone #13054

InderSethi opened this issue May 30, 2024 · 7 comments
Labels
bug Something isn't working Stale Stale and schedule for closing soon

Comments

@InderSethi
Copy link

Search before asking

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

YOLOv5 Component

Training

Bug

I m unable to train the model on visDrone Throws below error:

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/iiitd/Documents/inder/test.py", line 7, in
results = model.train()
^^^^^^^^^^^^^
File "/home/iiitd/Documents/inder/venv/lib/python3.12/site-packages/ultralytics/engine/model.py", line 655, in train
self.trainer = (trainer or self._smart_load("trainer"))(overrides=args, _callbacks=self.callbacks)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/iiitd/Documents/inder/venv/lib/python3.12/site-packages/ultralytics/engine/trainer.py", line 130, in init
self.trainset, self.testset = self.get_dataset()
^^^^^^^^^^^^^^^^^^
File "/home/iiitd/Documents/inder/venv/lib/python3.12/site-packages/ultralytics/engine/trainer.py", line 521, in get_dataset
raise RuntimeError(emojis(f"Dataset '{clean_url(self.args.data)}' error ❌ {e}")) from e
RuntimeError: Dataset 'VisDrone.yaml' error ❌ ❌ Download failure for https://github.com/ultralytics/yolov5/releases/download/v1.0/VisDrone2019-DET-train.zip. Environment is not online.I'm

Environment

I tried on google colab and on my own PC. A couple of days back it was working fine.

Minimal Reproducible Example

model = YOLO('https://hub.ultralytics.com/models/WbKCeFB1gOrXhGcK3xM5')
results = model.train()

Additional

No response

Are you willing to submit a PR?

  • Yes I'd like to help by submitting a PR!
@InderSethi InderSethi added the bug Something isn't working label May 30, 2024
@github-actions GitHub Actions
Copy link
Contributor

github-actions bot commented May 30, 2024 •
edited by UltralyticsAssistant
Loading

👋 Hello @InderSethi, 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):

Status

YOLOv5 CI

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
Copy link
Member

Hello,

Thank you for reporting this issue with training on the VisDrone dataset. From the error message, it appears that the dataset download failed because the environment was not online. This could be due to a temporary network issue or a problem with the dataset URL.

Please ensure that your internet connection is stable and try downloading the dataset manually from the provided URL to verify its availability. If the issue persists, it might be helpful to check if the dataset URL has changed or if there are any restrictions on downloading files in your current environment (such as firewall settings).

If you continue to face difficulties, please provide any additional details or changes in your setup since the last successful training. This will help us better understand and address the problem.

Thank you for your cooperation and patience.

@InderSethi
Copy link
Author

InderSethi commented May 31, 2024 via email

@glenn-jocher
Copy link
Member

@InderSethi hi Inder,

Thank you for the update and for pinpointing the issue with the URL. It seems like the extra period was indeed causing the download to fail. I appreciate your diligence in troubleshooting this problem.

We will review the dataset URLs in our documentation and scripts to ensure they are correct and prevent similar issues in the future. Meanwhile, your manual fix to remove the period and successfully download the dataset is a good interim solution.

If you encounter any further issues or have additional insights to share, please feel free to reach out.

@InderSethi
Copy link
Author

InderSethi commented May 31, 2024 via email

@glenn-jocher
Copy link
Member

Hi Inder,

To use an offline dataset with the YOLOv5 code from YOLO Hub, you can specify the path to your local dataset in the dataset configuration file (usually a YAML file). Here’s a quick guide:

  1. Ensure your dataset is structured correctly (images and labels in expected directories).
  2. Modify the dataset YAML file:
    • Replace the path value with the local path to your dataset.
    • Ensure train, val, and test paths are correctly set relative to the new path.
  3. Use this YAML file path when initializing your model training in the YOLO Hub code.

This should allow the model to train using your locally stored dataset. If you need further assistance, please let me know!

@github-actions GitHub Actions
Copy link
Contributor

github-actions bot commented Jul 1, 2024

👋 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:

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 ⭐

@github-actions github-actions bot added the Stale Stale and schedule for closing soon label Jul 1, 2024
@github-actions github-actions bot closed this as not planned Won't fix, can't repro, duplicate, stale Jul 12, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working Stale Stale and schedule for closing soon
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@InderSethi @glenn-jocher