Questions tagged [yolo]

For questions related to the family of models known as YOLO (which stands for "You Only Look Once"), which were proposed by Joseph Redmon et al. There are at least three YOLO models (versions 1, 2, and 3).

Filter by
Sorted by
Tagged with
2 votes
1 answer
64 views

Data scan not making sense for coco dataset

I am doing a simple scan to see how dataset size affects training. Basically, I took 10% of the coco dataset and trained a yolov3 net (from scratch) to just look for people. Then I took 20% of the ...
1 vote
1 answer
47 views

NN Architecture for the detection of "sparse" Objects

I have a document digitalization task where I want to detect technical drawings from images. These Images mostly consist of objects made up of combination of shapes like lines, circles and rectangles. ...
2 votes
1 answer
2k views

How does YOLO detect the object when the object is in multiple grid cells?

I have been reading various articles and watching videos on YouTube, but I can't seem to understand one thing. How does YOLO make a bounding box for an object if it is in multiple grid cells? For ...
1 vote
1 answer
42 views

How to automatically trigger model retraining for object detection models in case of data drift?

I have fine-tuned a pre-trained Yolov8 model on my dataset of labelled containers in a warehouse conveyor belt (image example here) . I am working to develop an MLOps project for my project portfolio, ...
4 votes
1 answer
1k views

How can I incrementally train a Yolo model without catastrophic forgetting?

I have successfully trained a Yolo model to recognize k classes. Now I want to train by adding k+1 class to the pre-trained weights (k classes) without forgetting previous k classes. Ideally, I want ...
0 votes
2 answers
112 views

How to detect negative (absence of) an object?

I want to detect the people that are NOT wearing PPE vests using a pre-trained object detection model like YOLO or Grounding Dino. The models are able to detect people and vests separately, but I am ...
0 votes
1 answer
67 views

Handle overlapping objects in instance segmentation annotation

As I struggle to find any literature online about this, I wanted to create a discussion that other would be able to learn from. My question is inspired from a Yolo GitHub issue. In this example we ...
0 votes
2 answers
775 views

For an image (of any object), how to find its location in the other image(s) which contains it, given there are no labels or annotations for any image

Problem Statement: I am given 2 sets of images. All the images in both sets are without annotations and labels. First set : a set of images of the grocery store shelves (captured in the grocery stores)...
2 votes
1 answer
613 views

How to detect multiple playing cards of the same class with a neural network?

I want to train an AI to detect the class (i.e. suit and rank) of playing cards. Playing cards from different decks may use slightly different shapes or colors to represent these attributes, and I ...
0 votes
0 answers
41 views

How does the Yolo loss match bounding boxes to the ground truth?

I've been going over the YOLO paper again and I was wondering something about the loss. Yolo divides an image into grids and then for each grid can have multiple bounding boxes to detect multiple ...
0 votes
2 answers
88 views

Performance of augmented dataset with or without original images

I am training on yolo and I had a small dataset. I decided to increase it by augmenting it with rotation, shearing, etc to increase the size and increase accuracy. Now I have seen augmented datasets ...
0 votes
2 answers
35 views

Should I define my problem as image segmentation or detection?

I have a problem and have to decide wether it's an object detection or object segmentation problem. I want to use Yolov8 for training. We already have hundrets of images but they aren't labeled yet. ...
0 votes
1 answer
354 views

Object Classification: How to decide which detected region is a RoI for classification?

I am working on a project where I am working on the Flickr-47 dataset to do logo detection and classification. My approach is to first finetune a YOLO v5 model with high recall to detect as many "...
7 votes
1 answer
2k views

What are the differences between Yolo v1 and CenterNet?

I recently read a new paper (late 2019) about a one-shot object detector called CenterNet. Apart from this, I'm using Yolo (V3) one-shot detector, and what surprised me is the close similarity between ...
-2 votes
1 answer
51 views

How to run our python scripts utilizing our device's GPU? [closed]

My laptop has NVIDIA GeForce GTX1650 GPU. I want to utilize this GPU to run my Python script. Any help in the form of code would be really helpful. I mean tried researching this so much but I couldn't ...
0 votes
1 answer
472 views

How to extract the high-level features of YOLOv5?

I am faimilar with extracting the high-level features from any pretrained model for classification problem such as ResNet version, VGG, etc. It is easy to extract the features because there is a fully ...
0 votes
0 answers
25 views

Training YOLO on YOLO results

I have a large unlabelled dataset. What if I use YOLO to label it? Will this dataset be useful to train a better YOLO model? What if I then finetune it on a smaller labelled dataset? My usecase ...
0 votes
0 answers
16 views

YoloV5 Calibration

I'm currently tackling the task of calibrating object detection models specifically tailored for medical imaging applications. Despite extensive online research, I've encountered difficulties in ...
0 votes
1 answer
621 views

Feeding YOLOv4 image data into LSTM layer?

How would one extract the feature vector from a given input image using YOLOv4 and pass that data into an LSTM to generate captions for the image? I am trying to make an image captioning software in ...
0 votes
0 answers
25 views

Yolo object detection metrics

I have made some predictions and saved the results to YOLO format. Then I made a program to calculate metrics, every metric looks fine except Precision/Confidence curve. I guess the flatline at the ...
0 votes
1 answer
43 views

YOLOv1 doesn't work for custom dataset

I am currently trying to train my own YOLOv1 network, based on this repository: https://github.com/ivanwhaf/yolov1-pytorch The images I want to train look like this: Three classes I want to detect: ...
0 votes
1 answer
135 views

Why YOLOv7 not detecting small objects

I am using YOLOv7 trained on custom dataset, and using the model weights after converting to ONNX for CVAT annotation. There it's only predicting few small objects. Is there any limit on small object ...
0 votes
1 answer
316 views

Yolov8 with key points detection without object detection

I have a project where the data is annotated with key points without any bounding boxes. If I want only the key points without the bounding box around the object, is that possible or each object ...
1 vote
1 answer
329 views

Autodistill: Train YOLOv8 with ZERO Annotations

I'm trying to create a distilled model of SAM https://colab.research.google.com/drive/1PKc5x1XjxOPP-2msFp6uMSULvnm9dorw?usp=sharing I'm able to label the data and train, but the predict doesn't find ...
0 votes
0 answers
63 views

How to track arbitrary object picked by mouse using YOLO?

We can track human, car, animal, etc using YOLO V8. In order to track object, the model has to be trained on target objects. But I have to track arbitrary objects picked by mouse on a video. Since the ...
0 votes
1 answer
150 views

Can DeepSort be made to track objects beside people?

As far as my understanding goes, the model used for feature extraction in DeepSort is specified as the first argument of the function create_box_encoder in the file ...
0 votes
0 answers
30 views

Unexpected improvements in model

I have a poor dataset, where of the training data has a single object labelled in the image. Same for the validation set even though I reserved the best (more general and well labelled) to then select ...
1 vote
0 answers
110 views

Should I label every frames from videos that I want to train my model to?

I want to build a model that can detect the chords that are being played on a guitar. Now I am planning to build it using YOLOv8 and annotate the frames using CVAT. Now I also have seen the tutorials ...
0 votes
0 answers
411 views

YOLOv7 fine tuning with unbalanced dataset

I have a question about training object detection models especially using the YOLOv7 algorithm. I use a Soccer Players Dataset from the Roboflow. The dataset is very unbalanced. I trained the model ...
0 votes
1 answer
353 views

Using GANs to generate data augmentations for YOLOv5

I was building a YOLOv5 object detection model, and was looking into researching synthetic methods like GANs to increase the size of my training set in an unsupervised manner. I know that few-shot ...
0 votes
1 answer
949 views

Split dataset into Train/Validation/Test for Object Detection

I have a dataset for Object Detection with YOLO format labels, each imagine can have occurences of different classes and multiple occurences of the same class. How can the dataset be divided into ...
3 votes
1 answer
2k views

What are the main differences between YOLOv3 and RetinaNet object detection algorithms?

I am looking at a certain project that compares performance on a certain dataset for an object detection problem using YOLOv3 and RetinaNet (or the "SSD_ResNet50_FPN" from TF Model Zoo). ...
2 votes
2 answers
220 views

How does YOLO handle non-class objects?

I have been reading more about computer vision and I'm bothered by YOLO and similar deep learning architectures. The thing I am confused about is how non-class image sections are dealt with. In ...
0 votes
1 answer
51 views

Taxonomy of terms in DL [closed]

I am trying to teach myself DL and among other difficulties I found it quite challenging to structurize a very basic terminology vocabulary especially when it includes not only theoretical stuff but ...
2 votes
2 answers
367 views

What YOLO algorithm can I use for images with noise as I will implement it in real time?

I want to detect drivers with or without seatbelts at crossroads. For that, as it is real-time, I am going to use the YOLO algorithm/model. For training data sets (the images) I need to collect, I ...
0 votes
1 answer
169 views

Can YOLOv3 architecture be clearly separated into feature extractor and classifier parts?

I am new to machine learning and am confused about whether its architecture has clearly defined boundaries which demarcate the feature extraction and classification part. Or is it that it classifies ...
0 votes
1 answer
102 views

Darknet as a part of Yolo v3

I am pretty new to ML and my question may look strange. Especially the last part of it. 1)As far as I understand Darknet53 is an integral part of Yolo just as Resnet50 is a part of R-CNN Am I right? 2)...
1 vote
0 answers
55 views

Training with extremely imbalanced Dataset

I have a object detection problem which has extremely imbalanced dataset. Lets say there is only one class to detect, say apple or not apple. This detection network will be used in a real case ...
0 votes
0 answers
145 views

What object detection algorithm is the best for my particular problem?

I am trying to write a program to put a bounding box around dead fish, and not the live ones, in a video. I have minimal data (~5k frames and ~7k objects in total ) and it is VERY low quality (poor ...
2 votes
3 answers
601 views

Would YOLO be able to detect objects in "different" positions?

I have the following question about You Only Look Once (YOLO) algorithm, for object detection. I have to develop a neural network to recognize web components in web applications - for example, login ...
1 vote
1 answer
49 views

How to identify and diferentiate several edge lines of an object?

I want to create an AI to detect and identify certain edge lines on my image. The input image is a locker key, and I want to know the exact position of certain edges. Sample input image: Sample ...
0 votes
1 answer
1k views

YOLO - does the Intersection over Union is actually a part of Non Maximum Suppresion

In the Stack Overflow thread Intersection Over Union (IOU) ground truth in YOLO they say that in YOLO actually the IoU (intersection over union) is used twice: during training to compare ground truth ...
3 votes
1 answer
447 views

YOLO - are the anchor boxes used only in training?

another question in YOLO. I've red about how YOLO adjusts anchor boxes by offsets to create the final bounding boxes. What I do not understand, is when YOLO does it. Is it being done only during the ...
2 votes
2 answers
2k views

How can I detect thin objects (like pens and pencils) without a bounding box but only 2 endpoints and the orientation?

I am looking to detect thin objects, like pens, pencils, and surgical instruments. The bounding box is not important, but I am looking to see if I can train a model to detect both the object as well ...
0 votes
1 answer
316 views

Are the output dimensions of the first and second convolutional layer in YOLO paper correct?

I was reading the last version of the YOLO paper available in Arxiv, and I don't fully understand the output dimensions (I understand width and height, but not depth) of the first and second ...
0 votes
1 answer
319 views

Multiple labels for the same rectbox?

My goal is to identify the horse in a photo. I'm dealing with about 500 unique horses. My feeling is that the best way to distinguish one horse from another is by its face. So I trained Yolov5 ...
1 vote
0 answers
85 views

Different equations for Yolov3 in courses/ articles and Darknet GitHub code?

I am confused by the equations for bounding boxes I find online. Some articles say that box_width = anchor_width * exp(residual_value_of_box_width)) and the ...
1 vote
0 answers
45 views

In anchor based object detection, why don't the anchors share the same weights?

After reading about YOLO V3 and Faster R-CNN, I don't understand why the weights for the regression head aren't the same across all boxes of the same size. Given that the backbone of these systems is ...
0 votes
1 answer
47 views

Should I label static objects on video dataset?

I'm using nvidia Transfer Learning Toolkit to detect cars in some video frames. I found some dataset (for example https://www.jpjodoin.com/urbantracker/dataset.html and https://www.kaggle.com/...
2 votes
1 answer
142 views

Preparing data set for the YOLO algorithm

Hi I am working on a project which requires the You Only Look Once algorithm in order to classify and localise objects within images. I have to prepare my dataset (which has 2 classes, and predicts 6 ...