Comments (4)
yolov5.py是什么文件?你自己写的吗
from tensorrtx.
是的我自己写的,我参照你们提供的模版去转化,发现唯独yolov5读取的时候直接报错了
就是类似使用python去实现这一段读取代码std::map<std::string, Weights> loadWeights(const std::string file)
{
std::cout << "Loading weights: " << file << std::endl;
std::map<std::string, Weights> weightMap;
// Open weights file
std::ifstream input(file);
assert(input.is_open() && "Unable to load weight file.");
// Read number of weight blobs
int32_t count;
input >> count;
assert(count > 0 && "Invalid weight map file.");
while (count--)
{
Weights wt{DataType::kFLOAT, nullptr, 0};
uint32_t size;
// Read name and type of blob
std::string name;
input >> name >> std::dec >> size;
wt.type = DataType::kFLOAT;
// Load blob
uint32_t* val = reinterpret_cast<uint32_t*>(malloc(sizeof(val) * size));
for (uint32_t x = 0, y = size; x < y; ++x)
{
input >> std::hex >> val[x];
}
wt.values = val;
wt.count = size;
weightMap[name] = wt;
}
return weightMap;
}
from tensorrtx.
但是我不知道这个报错怎么去解决,它似乎是读取问题,案例其实就是参照了,yolov5-v7 src 里面的model.cpp 和resnet50.py,然后去实现这些操作
from tensorrtx.
Related Issues (20)
- Fix issue in boxes calculation in 'rcnn/PredictorDecode.cu' HOT 2
- infer without results HOT 1
- INT8量化问题:最后输出模型时报错,提示"[E] [TRT] 1: Unexpected exception invalid unordered_map<K, T> key" HOT 4
- Nothing can be predicted after changing the code HOT 8
- yolov5 int8量化推理结果为空 HOT 6
- yolov8 TRT问题 HOT 5
- 从wts文件读取权重数据,应该是hex的float值, 为什么存储在 uint32_t中啊 HOT 2
- 为什么yolov8中,trtx的bottleneck实现与源码不同? 是有特殊的考虑吗?
- Are there any plans to implement support for YOLO-NAS? HOT 2
- Segmentation fault when generate engine HOT 2
- NameError: name 'false' is not defined when I try to generate .wts HOT 2
- Request for Dockerfile of Tensorrtx Image HOT 2
- Depedency Issue : Mxnet-Numpy HOT 4
- The yolov8n inference and post-processing time increases at the time of processing up to the 20th image and remains HOT 3
- yolov5 编译错误 HOT 2
- serialize engine from yolov8l-seg.wts failed! HOT 1
- yolov8s在960x960输入size存在较大概率误检,同一个模型用onnx或者torch推理正常的 HOT 6
- retinaface序列化时的报错 HOT 3
- 尝试修改输入分辨率后,pt转wts报错 HOT 6
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 tensorrtx.