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
0
votes
0 answers

What is the relation between arguments used in YAML and class arguments in the yolov5 implementation by ultralytics?

I was working on fish detection and tracking system in 3 dimensions using MS Kinect, and used yolov5s for rgb frames. I wanted to experiment a bit with the architecture of yolov5, but as a beginner i am struggling to find the relation between the…
0
votes
1 answer

Attribute Error while running train.py in YOLOv5

When I run python train.py --img 640 --batch 4 --epochs 5 --data training/dataset.yaml --cfg training/yolov5l.yaml --weights yolov5l.pt for YOLOv5 in my system I get the following error: Traceback (most recent call last): File "train.py", line…
Sandra
  • 11
  • 1
  • 3
0
votes
1 answer

Safety hat and vest detection with Yolo v4

I am training a Yolo v4 object detection model that detects the wearing of safety hats and vest on construction sites, I want to show as a result if the person is wearing the hat and vest or not, The bounding box of the person must show "person…
0
votes
0 answers

Yolov5 is not detecting the labels

I am training my model to detect the vertebra but after training, it shows no result. Is there any way I can get the results by tuning any of the parameters?
MakM
  • 27
  • 5
0
votes
0 answers

Image labelling, How to annotate a picture which don't have an object but i should have its xml file as i explained in the question

I find a dataset which some of the pictures is not annotated but it has like this format in xml. But this another picture has an annotated object which look like this: Can any one tell me How and by which tools i can annotated a picture which…
0
votes
2 answers

Logging training metrics to a csv file

I want to log all training metrics to a csv file while it is training on YOLOV5 which is written with pytorch but the problem is that I don't want to use tensorboard. To achieve this goal I tried some techniques like below: -First log it into…
kuti
  • 21
  • 3
0
votes
3 answers

How to use tensorRT in Yolov5?

I want to optimize my trained model in Yolov5 using tensorRT. But am unable to find a proper way of doing so.Can any body tell show me how to do so?
Bigyan Subedi
  • 99
  • 2
  • 9
0
votes
2 answers

convert yolov5s.pt to openvino IR format success, predict stuff failed

This is begin from curious....I download the pretrained yolov5s.pt from public google drive, and convert it as yolov5s.onnx file with input shape [1,3,640,640] by using yolov5's models/export.py. Then I use openvino's deployment tools/mo.py to…
Fenix Lam
  • 386
  • 6
  • 22
0
votes
1 answer

Train deep learning model in Amazon EC2 is extremely slow

I am having big speed troubles training YOLOv5 into my p2.xlarge aws ec2 instance which has a NVIDIA Tesla K80. I realized the training process was even slower than my desktop PC who has a NVIDIA RTX 2060. So I decided to inference over some images…
0
votes
0 answers

Yolov5 How to find pixel size of objects detected?

I'm looking to make yolov5 detect.py output the size of the objects found inside the bounding boxes. However, I cannot find the correct variable, can anyone help? Thanks in advance! link to code in github :…
0
votes
1 answer

Cover image for neural network training

I want to train an object detection model to detect chickens that are only upon that water reservoir (like the one on the picture), thus I'm currently annotating chickens in images. Since I don't want to detect any chickens on the field, I'm not…
João Gondim
  • 101
  • 1
  • 1
  • 9
0
votes
1 answer

I am using a Yolov5 pretrained model, I have successfully created yolov5s. Engine file, but something problem in yolov5_trt.py file. Core dumped

I follow all instruction which mentioned (https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5). Getting error while running python yolov5_trt.py file. python: engine.cpp:1104: bool nvinfer1::rt::Engine::deserialize(const void*, std::size_t,…
Yogi
  • 1
0
votes
1 answer

Why am I unable to train YOLOv5 on my dataset?

I'm trying to train the YOLOv5 model on a Jupyter notebook using a custom dataset. The dataset is a face mask detection dataset, containing images of people with/without facemasks. I've converted the annotations to YOLO format, and I believe I've…
user123965
  • 169
  • 8
0
votes
1 answer

Training custom model

I am trying to train my dataset on yolov5 I normalized data as discussed in the docs on github but I always end up with this error. from n params module arguments 0 …
0
votes
1 answer

Saving YOLOv3 Model State and Loading

I am trying to train YOLOv3 model following ImageAi documentation with my custom dataset.But the training is taking too much time that Google Colab time is not enough for this. Now how can I save Model state and load it after 50/60 epoch completion?…
Tech-D
  • 21
  • 2
  • 3