Stack Overflow
Stack Overflow

Questions tagged [yolov5]

YOLOv5 is a family of object detection architectures and models pretrained on the COCO dataset.

It is a product of Ultralytics open-source research.

643 questions
7
votes
1 answer

Process output data from YOLOv5 TFlite

❔Question Hi, I have successfully trained a custom model based on YOLOv5s and converted the model to TFlite. I feel silly asking, but how do you use the output data? I get as output: StatefulPartitionedCall: 0 = [1,25200,7] from the converted…
Tommydw
  • 181
  • 2
  • 8
6
votes
1 answer

YOLOV8 how does it handle different image sizes

Yolov8 and I suspect Yolov5 handle non-square images well. I cannot see any evidence of cropping the input image, i.e. detections seem to go to the enge of the longest side. Does it resize to a square 640x604 which would change the aspect ratio of…
Alan Johnstone
  • 455
  • 1
  • 7
  • 24
5
votes
4 answers

Read every nth frame from youtube live video for object detection

I am currently using a custom trained yolov5 model to run object detection inference on live youtube videos, the problem is that the videos are streamed at 30 FPS , I actually don't want to process each frame for object detection and just process…
Abhishake A
  • 59
  • 2
5
votes
3 answers

How to set and get confidence threshold from custom YOLOv5 model?

I am trying to perform inference on my custom YOLOv5 model. The official documentation uses the default detect.py script for inference. Example: python detect.py --source data/images --weights yolov5s.pt --conf 0.25 I have written my own python…
TechXpert
  • 155
  • 2
  • 3
  • 10
5
votes
2 answers

How to get class and bounding box coordinates from YOLOv5 predictions?

I am trying to perform inference on my custom YOLOv5 model. The official documentation uses the default detect.py script for inference. I have written my own python script but I cannot access the predicted class and the bounding box coordinates from…
TechXpert
  • 155
  • 2
  • 3
  • 10
5
votes
3 answers

Yolo V5 issue "Exception: Dataset not found." on local machine

I am trying to train a model using Yolo V5. I have the issue of Data base not found. I have a train, test and valid files that contain all the image and labels files. I have tested the files on googlecolap and it dose work. However, on my local…
Zag.D
  • 59
  • 1
  • 1
  • 4
4
votes
1 answer

why am I getting AssertionError: train: No labels found in /content/dataset/test/labels.cache, can not start training

AssertionError: train: No labels found in /content/dataset/test/labels.cache, can not start training Automatically generated cache files I'm doing yolo5 An error occurs during the configuration and training a dataset. I understand that the command…
김민재
  • 41
  • 1
  • 3
4
votes
4 answers

How to load custom yolo v-7 trained model

How do I load a custom yolo v-7 model. This is how I know to load a yolo v-5 model : model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolov5/runs/train/exp15/weights/last.pt', force_reload=True) I saw videos online and they suggested to…
pavan
  • 102
  • 2
  • 11
4
votes
0 answers

How to combine (results) of multiple pytorch models?

community, in pytorch (& yolov5) i was able to train two models (one detects cars, the other detects bikes). I am loading two different models (last.pt files) with the following commands: model_c = torch.hub.load('ultralytics/yolov5', 'custom',…
Phil Harmony
  • 41
  • 1
  • 2
4
votes
4 answers

ImportError: cannot import name 'find_stack_level' from 'pandas.util._exceptions'

When I tried to run the YOLOv5 train.py, I don't know what was wrong with it. pandas is installed. Traceback (most recent call last): File "/home/jasmine/Desktop/fyp/yolov5/yolov5/utils/general.py", line 26, in import pandas as pd …
jasssss
  • 41
  • 1
  • 1
  • 2
4
votes
1 answer

Yolov5 OpenCV error: (-215:Assertion failed) whilst using .onnx

I've just created a yolov5 model, and exported it in the onnx format so it is usable with opencv but I keep getting the error: [ERROR:0] global D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp (3554)…
Matthew Bevilacqua
  • 41
  • 1
  • 2
4
votes
2 answers

TypeError: load() missing 1 required positional argument: 'Loader'

I am trying to run this github repo found at this link: https://github.com/HowieMa/DeepSORT_YOLOv5_Pytorch After installing the requirements via pip install -r requirements.txt. I am running this in a python 3.8 virtual environment, on a dji…
fatbringer
  • 348
  • 2
  • 9
  • 19
4
votes
3 answers

How do I draw bounding boxes from "results.xyxy[0]" with cv2 rectangle (YOLOv5)?

New to both python and machine learning. I'm trying to draw bounding boxes on my mss screen capture. This is the part of the code where I believe I should be receiving the coordinates to draw the rectangle. while True: img =…
freepcvirusremoval
  • 41
  • 1
  • 2
4
votes
2 answers

How do I use the exported 'best.pt" file from yolov5 colab file to run the trained weights locally?

I have trained my model using yoloV5 on google colab, following the provided tutorial and walkthrough provided for training any custom model: Colab file for training your own custom model. I now have an exported best.pt file after running the last…
user16013401
3
votes
1 answer

How do I save a yolov5 model among the registered models in MLflow and then load it either directly or from weights (.pt) to make predictions?

My objective is to : save the yolov5 model in MLflow Model Registry Load the model directly from model registry so that we can make predictions. Have access to the results of prediction (bounding boxes). Current steps followed: Cloned the yolov5…
atharva parikh
  • 41
  • 3
1
2 3
…
42 43