search

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API
MIT License
7.04k stars 1.78k forks source link

yolov5: /home/air/yolo5_trt/tensorrtx/yolov5/yolov5.cpp:63: nvinfer1::ICudaEngine* build_engine(unsigned int, nvinfer1::IBuilder*, nvinfer1::IBuilderConfig*, nvinfer1::DataType, float&, float&, std::__cxx11::string&): Assertion `upsample11' failed. Aborted #958

Closed wxthu closed 2 years ago

wxthu commented 2 years ago

when i run sudo ./yolov5 -s yolov5x.wts yolov5x.engine s

image I was very confused

wang-xinyu commented 2 years ago

for model x, ./yolov5 -s yolov5x.wts yolov5x.engine x

wxthu commented 2 years ago

for model x, ./yolov5 -s yolov5x.wts yolov5x.engine x

Thanks， it seems that converting wts file to tensorrt engine takes long long time for yolov5x model and my computer even was stuck .... Is that correct or normal?

wxthu commented 2 years ago

image stuck for long long time。。。。

1025888636 commented 2 years ago

hey,the same problem i met,did you solved it?

1025888636 commented 2 years ago

i solved it，its because my model trained by yolov5x.pt，but sudo ./yolov5 -s yolov5x.wts yolov5x.engine s，it should sudo ./yolov5 -s yolov5x.wts yolov5x.engine x

stale[bot] commented 2 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.