Comments (9)
mihuzz
commented on May 14, 2024
Hi. You try to convert default yolov7-tiny model?
from tensorrt-for-yolo-series.
mihuzz
commented on May 14, 2024
for the first i see you have installed torch 1.12 but see requierments.txt
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt
Yes, here you can see that I'm using their default model:
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
I had previously installed all the requirements.
Running pip3 install -r requirements.txt outputs the following. As you can see @mihuzz , all requirements are met:
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
Requirement already satisfied: matplotlib>=3.2.2 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 4)) (3.5.3)
Requirement already satisfied: numpy>=1.18.5 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 5)) (1.23.1)
Requirement already satisfied: opencv-python>=4.1.1 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 6)) (4.6.0.66)
Requirement already satisfied: Pillow>=7.1.2 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 7)) (9.2.0)
Requirement already satisfied: PyYAML>=5.3.1 in /usr/lib/python3/dist-packages (from -r requirements.txt (line 8)) (5.3.1)
Requirement already satisfied: requests>=2.23.0 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 9)) (2.28.1)
Requirement already satisfied: scipy>=1.4.1 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 10)) (1.9.0)
Requirement already satisfied: torch!=1.12.0,>=1.7.0 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 11)) (1.12.1)
Requirement already satisfied: torchvision!=0.13.0,>=0.8.1 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 12)) (0.13.1)
Requirement already satisfied: tqdm>=4.41.0 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 13)) (4.64.1)
Requirement already satisfied: protobuf<4.21.3 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 14)) (3.19.4)
Requirement already satisfied: tensorboard>=2.4.1 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 17)) (2.9.1)
Requirement already satisfied: pandas>=1.1.4 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 21)) (1.4.3)
Requirement already satisfied: seaborn>=0.11.0 in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 22)) (0.11.2)
Requirement already satisfied: ipython in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 34)) (8.3.0)
Requirement already satisfied: psutil in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 35)) (5.9.0)
Requirement already satisfied: thop in /home/cyrus/.local/lib/python3.8/site-packages (from -r requirements.txt (line 36)) (0.1.1.post2209072238)
Requirement already satisfied: kiwisolver>=1.0.1 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (1.4.4)
Requirement already satisfied: cycler>=0.10 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (0.11.0)
Requirement already satisfied: packaging>=20.0 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (21.3)
Requirement already satisfied: fonttools>=4.22.0 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (4.36.0)
Requirement already satisfied: python-dateutil>=2.7 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (2.8.2)
Requirement already satisfied: pyparsing>=2.2.1 in /home/cyrus/.local/lib/python3.8/site-packages (from matplotlib>=3.2.2->-r requirements.txt (line 4)) (3.0.7)
Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/lib/python3/dist-packages (from requests>=2.23.0->-r requirements.txt (line 9)) (1.25.8)
Requirement already satisfied: idna<4,>=2.5 in /usr/lib/python3/dist-packages (from requests>=2.23.0->-r requirements.txt (line 9)) (2.8)
Requirement already satisfied: charset-normalizer<3,>=2 in /home/cyrus/.local/lib/python3.8/site-packages (from requests>=2.23.0->-r requirements.txt (line 9)) (2.1.1)
Requirement already satisfied: certifi>=2017.4.17 in /usr/lib/python3/dist-packages (from requests>=2.23.0->-r requirements.txt (line 9)) (2019.11.28)
Requirement already satisfied: typing-extensions in /home/cyrus/.local/lib/python3.8/site-packages (from torch!=1.12.0,>=1.7.0->-r requirements.txt (line 11)) (4.1.1)
Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (1.8.1)
Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (0.4.6)
Requirement already satisfied: grpcio>=1.24.3 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (1.47.0)
Requirement already satisfied: setuptools>=41.0.0 in /usr/lib/python3/dist-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (45.2.0)
Requirement already satisfied: google-auth<3,>=1.6.3 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (2.9.0)
Requirement already satisfied: werkzeug>=1.0.1 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (2.0.3)
Requirement already satisfied: wheel>=0.26 in /usr/lib/python3/dist-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (0.34.2)
Requirement already satisfied: markdown>=2.6.8 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (3.3.7)
Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (0.6.1)
Requirement already satisfied: absl-py>=0.4 in /home/cyrus/.local/lib/python3.8/site-packages (from tensorboard>=2.4.1->-r requirements.txt (line 17)) (1.1.0)
Requirement already satisfied: pytz>=2020.1 in /home/cyrus/.local/lib/python3.8/site-packages (from pandas>=1.1.4->-r requirements.txt (line 21)) (2022.2.1)
Requirement already satisfied: matplotlib-inline in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (0.1.3)
Requirement already satisfied: pygments>=2.4.0 in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (2.11.2)
Requirement already satisfied: prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0 in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (3.0.29)
Requirement already satisfied: traitlets>=5 in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (5.2.0)
Requirement already satisfied: pexpect>4.3; sys_platform != "win32" in /usr/lib/python3/dist-packages (from ipython->-r requirements.txt (line 34)) (4.6.0)
Requirement already satisfied: stack-data in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (0.2.0)
Requirement already satisfied: backcall in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (0.2.0)
Requirement already satisfied: pickleshare in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (0.7.5)
Requirement already satisfied: decorator in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (5.1.1)
Requirement already satisfied: jedi>=0.16 in /home/cyrus/.local/lib/python3.8/site-packages (from ipython->-r requirements.txt (line 34)) (0.18.1)
Requirement already satisfied: six>=1.5 in /usr/lib/python3/dist-packages (from python-dateutil>=2.7->matplotlib>=3.2.2->-r requirements.txt (line 4)) (1.14.0)
Requirement already satisfied: requests-oauthlib>=0.7.0 in /home/cyrus/.local/lib/python3.8/site-packages (from google-auth-oauthlib<0.5,>=0.4.1->tensorboard>=2.4.1->-r requirements.txt (line 17)) (1.3.1)
Requirement already satisfied: rsa<5,>=3.1.4; python_version >= "3.6" in /home/cyrus/.local/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r requirements.txt (line 17)) (4.8)
Requirement already satisfied: pyasn1-modules>=0.2.1 in /home/cyrus/.local/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r requirements.txt (line 17)) (0.2.8)
Requirement already satisfied: cachetools<6.0,>=2.0.0 in /home/cyrus/.local/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r requirements.txt (line 17)) (5.2.0)
Requirement already satisfied: importlib-metadata>=4.4; python_version < "3.10" in /home/cyrus/.local/lib/python3.8/site-packages (from markdown>=2.6.8->tensorboard>=2.4.1->-r requirements.txt (line 17)) (4.11.3)
Requirement already satisfied: wcwidth in /home/cyrus/.local/lib/python3.8/site-packages (from prompt-toolkit!=3.0.0,!=3.0.1,<3.1.0,>=2.0.0->ipython->-r requirements.txt (line 34)) (0.2.5)
Requirement already satisfied: pure-eval in /home/cyrus/.local/lib/python3.8/site-packages (from stack-data->ipython->-r requirements.txt (line 34)) (0.2.2)
Requirement already satisfied: asttokens in /home/cyrus/.local/lib/python3.8/site-packages (from stack-data->ipython->-r requirements.txt (line 34)) (2.0.5)
Requirement already satisfied: executing in /home/cyrus/.local/lib/python3.8/site-packages (from stack-data->ipython->-r requirements.txt (line 34)) (0.8.3)
Requirement already satisfied: parso<0.9.0,>=0.8.0 in /home/cyrus/.local/lib/python3.8/site-packages (from jedi>=0.16->ipython->-r requirements.txt (line 34)) (0.8.3)
Requirement already satisfied: oauthlib>=3.0.0 in /usr/lib/python3/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tensorboard>=2.4.1->-r requirements.txt (line 17)) (3.1.0)
Requirement already satisfied: pyasn1>=0.1.3 in /home/cyrus/.local/lib/python3.8/site-packages (from rsa<5,>=3.1.4; python_version >= "3.6"->google-auth<3,>=1.6.3->tensorboard>=2.4.1->-r requirements.txt (line 17)) (0.4.8)
Requirement already satisfied: zipp>=0.5 in /home/cyrus/.local/lib/python3.8/site-packages (from importlib-metadata>=4.4; python_version < "3.10"->markdown>=2.6.8->tensorboard>=2.4.1->-r requirements.txt (line 17)) (3.8.0)
from tensorrt-for-yolo-series.
Linaom1214
commented on May 14, 2024
@cyrusbehr @mihuzz
update your torch version, you can use the newest version.
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
@Linaom1214 I am apparently already using the newest version
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
@Linaom1214 is there a different version I should be using?
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
I followed the instructions in the colab notebook exactly. I'm not sure why it's not working...
from tensorrt-for-yolo-series.
cyrusbehr
commented on May 14, 2024
Managed to get the conversion working in a nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu20.04 container, indicated that there was indeed something wrong with my local env.
from tensorrt-for-yolo-series.
Related Issues (20)
- YOLOv7 Tensorrt converted model inference is equal to PyTorch model HOT 3
- int8 vs fp16 加速倍数能有多少？ HOT 1
- How to use engine in a process or a thread HOT 4
- how to deploy in multiple nvidia card, such as a computer with 8 3060 card?
- Add dynamic batch support for converting from onnx to .engine?
- auto in_dims = engine->getBindingDimensions(engine->getBindingIndex("image_arrays")); HOT 1
- En715 Jetson xaiver Nx Yolov7.trt Not detect HOT 2
- yolov7，official，int8，onnx-> trt报错 HOT 3
- c++ endtoend 关于预测的置信度绘制 HOT 4
- memory leak: Destroy function does not work
- Detection duplicates with fp16 on Jetson Nano (TensorRT v8.2.1.8) HOT 2
- Support for windows?
- License? HOT 4
- 关于V8 tensorrt 出现乱框的情况 HOT 33
- TensorRT Conversion Issue "TypeError: pybind11::init(): factory function returned nullptr" HOT 2
- about tensort version question or other question？ HOT 5
- yolov8 inference HOT 3
- Error Code 1: Serialization (Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry) HOT 2
- wrong confidence score (negative confidence score) on Jetson Nano inference HOT 3
- usage example for image_batch.py 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 tensorrt-for-yolo-series.