Comments (3)
I export the pt to onnx,then convert then onnx to tensorrt model, the use the tensorrt model to forward, all yolov5 and yolov6 is ok, but when yolov7 forward,
const int inputIndex = engine.getBindingIndex(INPUT_BLOB_NAME); //return -1
const int outputIndex = engine.getBindingIndex(OUTPUT_BLOB_NAME); //return -1
from yolov7.
output_names=['classes', 'boxes'] if y is None else ['output']
Maybe this? Classes and boxes
from yolov7.
I test with classes and boxes, still return -1
from yolov7.
Related Issues (20)
- question about ComputeLossOTA
- Hh
- onnx non end2end
- Problem wher run learning on 32 batch-size
- What exactly does image size do in test.py and detect.py?
- Change class weights HOT 2
- Memory Leak with PyTorch 2.3.1 and torchvision 0.18.1
- Using the train_aux.py training script, single gpu training fails, but multiple gpu training succeeds.
- Explaining this model with grad-CAM
- During the tranining dataloader problem HOT 1
- clearml.backend_api.session.defs.MissingConfigError: It seems ClearML is not configured on this machine!
- merging of two trained yolov7-tiny model with different classes
- Best Python version? HOT 1
- No pretrained model 'yolov7-tiny-seg.pt' to download
- It make '00123.jpg' to {"image_id": 123, when save json HOT 1
- Dear author, I have a question about cfg
- --patience argument missing
- No Detection of images even after training 1200 images with 100 epochs.
- No Detection of images even after training 1200 images with 100 epochs.
- paper
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 yolov7.