Comments (5)
Could you share your notebook,
so I can debug through your operations?
from onnx_tflite_yolov3.
Thank you for your reply.
It is difficult for me to share the notebook. I can tell you the steps I followed:
#Installation
!pip install onnx==1.6.0 onnx-tf==1.5.0
!pip install tensorflow-gpu==1.15.0
#Cloned your repository
!git clone https://github.com/zldrobit/onnx_tflite_yolov3
#Uploaded YOLOv3 .cfg and .weights files and changed the directory to onnx_tflite_yolov3 folder
#And then, when I ran this command on Colab:
!python3 detect.py --cfg cfg/yolov3-counter_detection.cfg --weights weights/yolov3_counter_detection.weights
The error occurred during its execution.
from onnx_tflite_yolov3.
Here's my notebook (https://colab.research.google.com/drive/1Cm3sQXoBMOnCFYNNKqAS-5eqJdlj8gV5?usp=sharing) to reproduce the python3 detect.py xxxxxxx
procedure.
This notebook runs without any error.
You could compare your notebook with this one to find out the cause of the problem.
from onnx_tflite_yolov3.
I close this issue since there is no recent activity.
from onnx_tflite_yolov3.
Hi, facing same problem converting my yolov3 darknet model. Exactly the same error message with your colab notebook.
This error does not occur with the base yolov3 weights but with a custom single-class model.
@zldrobit any idea ? @shreyapamecha did you solve it ?
File "/content/drive/My Drive/onnx_tflite_yolov3/models.py", line 203, in forward
p_cls = torch.cat(p_cls0, io[..., 6:], -1)
TypeError: cat() received an invalid combination of arguments - got (int, Tensor, int), but expected one of:
- (tuple of Tensors tensors, name dim, Tensor out)
didn't match because some of the arguments have invalid types: (int, Tensor, int) - (tuple of Tensors tensors, int dim, Tensor out)
didn't match because some of the arguments have invalid types: (int, Tensor, int)
from onnx_tflite_yolov3.
Related Issues (12)
- pytorch和tensorflow的cudnn冲突? HOT 4
- Can you provide some details about these lines in yolov5_tflite HOT 2
- The detailed commands for building TFLite GPU delegate so files HOT 3
- An Issue when loading OpenGL backend TFLite GPU delegate in Python HOT 3
- 镜像环境运行报错 HOT 3
- Segmentation fault(core dump) HOT 2
- Converting unsupported operation: PyFunc HOT 1
- Use of Reshape feature HOT 4
- Error while trying to convert custom built yolov3 model to tflite model HOT 17
- Segmentation fault (core dumped) in detect.py HOT 2
- Doesn't work with anything remotely recent. HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from onnx_tflite_yolov3.