python (12.9k questions)

javascript (9.2k questions)

reactjs (4.7k questions)

java (4.2k questions)

java (4.2k questions)

c# (3.5k questions)

c# (3.5k questions)

html (3.3k questions)

Questions - yolov5

TensorFlow Lite model not working in Android (Java) but works on Python

Background I successfully converted my YOLOV5 model from YOLOV5.pt to YOLOV5-fp16.tflite and was able to run it on python(refer to the test video on Google Drive). I used the conversion method python ...
test-img

NYProject

java

android

android-studio

tensorflow-lite

yolov5

Votes: 0

Answers: 1

Latest Answer

Your model output has shape [1, 6300, 6] not [1, 10, 4]. You will need to know what is the output of your model means and adjust the code correctly.
test-img

Karim Nosseir

YOLOv5 only detects when changing video resolution

I'm using YOLOv5 to detect persons in a video, which has a 944x1080 resolution. I'm using OBS-studio to stream the video as a virtual webcam. This is for testing purposes, but with the idea of using r...
test-img

schrodingerscatcuriosity

opencv

pytorch

resolution

yolov5

Votes: 0

Answers: 1

Latest Answer

In detect.py you have to resize the tensor with pytorch: from torchvision.transforms import Resize as rs im = torch.from_numpy(im).to(device) # line 119 r = rs((736, 1280)) i...
test-img

schrodingerscatcuriosity

Cuda illegal memory access when running inference on *.engine

After exporting a YoloV5 model to .engine I receive an error when trying to perform inference on it. Loading model.engine for TensorRT inference... [01/16/2022-04:18:26] [TRT] [I] [MemUsageChange] Ini...
test-img

user17948924

tensorflow

pytorch

tensorrt

yolov5

Votes: 0

Answers: 1

Latest Answer

Yes this is a known YOLOv5 TRT issue with AutoShape. The following code works correctly: python export.py --weights yolov5s.pt --include engine python detect.py --weights yolov5s.engine But if we use...
test-img

Glenn Jocher

How to convert yolov5 trained model into tflite in google colab to install on mobile app?

I have used Yolov5 to train the model in google colab and how can I export the .tflite file of the trained model so that it can be used for mobile applications. Thanks for help me
test-img

Nguyễn Khang

python

google-colaboratory

tensorflow-lite

training-data

yolov5

Votes: 0

Answers: 2

Latest Answer

Assuming that you have the latest TensorFlow version 2.7.0 use the general method for that purpose that it is shown here: import tensorflow as tf # Convert the model converter = tf.lite.TFLiteConvert...
test-img

Farmaker

Posts

Questions

Blogs

Jobs