YOLOv5 is a family of object detection architectures and models pretrained on the COCO dataset.
It is a product of Ultralytics open-source research.
Questions tagged [yolov5]
643 questions
2
votes
3 answers
YOLOV5 Despite having enough vram, I get RuntimeError: CUDA error: out of memory error even though there is not enough vram
I am traning YOLOV5 "L6" model for my important project. I have a so huge dataset contains UAV and drone image, and I need the train with huge input dimension (A few months ago I train "M" model with 640x640 input dimension with RTX 3060) in the…
Tony Stark
- 25
- 3
2
votes
1 answer
YoloV8 Segmentation mask problem (masks looks 2 parts but outputs only first part)
My code is as shown below:
import numpy as np
from ultralytics import YOLO
from PIL import Image
import cv2
from google.colab.patches import cv2_imshow
model = YOLO("yolov8n-seg.pt")
results = model.predict("payload.jpeg")
masks =…
jawamba
- 21
- 4
2
votes
1 answer
Specify a class to detect using YOLOv8 on pre-trained model
I'm new to YOLOv8, I just want the model to detect only some classes, not all the 80 classes the model trained on. How can I specify YOLOv8 model to detect only one class? For example only person.
from ultralytics import YOLO
model =…
TAN
- 23
- 1
- 7
2
votes
0 answers
Subprocess can't find module when not run with shell=True (Python)
When I use Popen to start a subprocess, if shell=True isn't set I get the following error message:
Traceback (most recent call last):
File "C:\Users\Student01\PycharmProjects\yolov5-master\detect.py", line 37, in
import…
Great God gaming
- 21
- 3
2
votes
0 answers
is there a way to convert a yolov5 dataset for rcnn or a mask-rcnn?
I currently got a yolov5 dataset , with everything on it (labels in form of : label , x , y , widh , height).
My question is , is there an fast way to convert it into a proper custom dataset for mask-rcnn?
Im currently searching the whole web,…
Murphy
- 21
- 2
2
votes
1 answer
what is the correct strides when add decoding level to a mlmodel converted from yolov7
I've trained a custom model with yolov7 and converted it to mlmodel with export.py (from yolov7 repo: https://github.com/WongKinYiu/yolov7).
Since export mlmodel's missing metadata, I was trying to add a decode layer to it, based on this post:…
Ray
- 29
- 2
2
votes
2 answers
Yolov5: How to combine 2 datasets (COCO + custom) when training
I want to train yolov5 by combining the coco dataset and the custom dataset created with roboflow. How do I merge datasets?
NSA
- 21
- 2
2
votes
1 answer
How can I tell Yolov5 to ignore Weights and Biases, ClearML, and other 3rd party apps?
When doing custom model training with Yolov5, the modeling framework seems to feel that I should be using Weights and Biases (wandb), ClearML, TensorBoard, and other external MLOps applications that I don't need and don't want to see excess messages…
Hack-R
- 22,422
- 14
- 75
- 131
2
votes
0 answers
Why do my results differ using the same weights and same network?
I want to use YOLOv5 pre-trained layer weights as part of a multi-task network. For this I saved the state_dict, loaded my custom model and overwrote the object detection layers (see below). Simply calling model.load_state_dict does not work, since…
Toni
- 21
- 3
2
votes
1 answer
How to customize data loader of yolov5 to train on VisDrone dataset?
I am new to deep learning. I am doing a school project where I am trying to train a YOLOv5 model on VisDrone dataset. My training set has 4911 images and validation set has more than 3000 images but less than 4000. I am using google Colab pro. As…
ShehjadK
- 31
- 7
2
votes
1 answer
Is there any way to use two different weight file of Yolov5 for a video?
I have two trained models of YoloV5 for pen and pen cap detection (pen.pt, cap.pt).
I want to use both this model for a video. For that I run a command
! python detect.py --weights cap.pt pen.pt --img 640 --conf 0.50 --source…
sourab maity
- 1,025
- 2
- 8
- 16
2
votes
1 answer
How to label objects when using YoloV5 for fresh/rotten fruits?
So I wanted to create a model that can classify a fruit into either "fresh" or "rotten". Currently I used this dataset from kaggle - https://www.kaggle.com/datasets/sriramr/fruits-fresh-and-rotten-for-classification
I then used LabelImg to apply…
newprogrammer2000
- 21
- 1
2
votes
0 answers
total(srcShape, srcRange.start, srcRange.end) == maskTotal in function 'computeShapeByReshapeMask'
I get the following error when trying to test the yolov5.onxx model with opencv
File "yolov5_opnecv.py", line 119, in
outs = detect(inputImage, net)
File "yolov5_opnecv.py", line 28, in detect
preds = net.forward()
cv2.error:…
Mohit Gaikwad
- 140
- 9
2
votes
1 answer
Running yolov5 with webcam inside Docker
I'm trying to do inference with a yolov5 model on the webcam stream from a Nvidia Jetson Nano within a Docker container. On both, my Windows PC and Jetson, I get the error:
qt.qpa.xcb: could not connect to display
qt.qpa.plugin: Could not load the…
tintinnabulum
- 63
- 5
2
votes
1 answer
What is this error in Pytorch model " AttributeError: 'DetectMultiBackend' object has no attribute 'input_details'?
I am running Yolov5 Model in flask file in AWS EC2 instance, Getting this weird error "DetectMultiBackend has no input_detail attribute.
I am using this to load the Yolov5 Pytorch Model
model = torch.hub.load('.', 'custom', path='best.pt',…
Amit Yadav
- 21
- 2