nvidia@tegra-ubuntu:~/program/deepstream_reference_apps-DS2/sources/apps/trt-yolo$ trt-yolo-app
File does not exist : /home/nvidia/program/deepstream_reference_apps-DS2/sources/lib/models/yolov2-tiny-kFLOAT-batch1.engine
Unable to find cached TensorRT engine for network : yolov2-tiny precision : kFLOAT and batch size :1
Creating a new TensorRT Engine
Loading pre-trained weights...
Loading complete!
Total Number of weights read : 11067854
layer inp_size out_size weightPtr
(1) conv-bn-leaky 3 x 416 x 416 16 x 416 x 416 496
(2) maxpool 16 x 416 x 416 16 x 208 x 208 496
(3) conv-bn-leaky 16 x 208 x 208 32 x 208 x 208 5232
(4) maxpool 32 x 208 x 208 32 x 104 x 104 5232
(5) conv-bn-leaky 32 x 104 x 104 64 x 104 x 104 23920
(6) maxpool 64 x 104 x 104 64 x 52 x 52 23920
(7) conv-bn-leaky 64 x 52 x 52 128 x 52 x 52 98160
(8) maxpool 128 x 52 x 52 128 x 26 x 26 98160
(9) conv-bn-leaky 128 x 26 x 26 256 x 26 x 26 394096
(10) maxpool 256 x 26 x 26 256 x 13 x 13 394096
(11) conv-bn-leaky 256 x 13 x 13 512 x 13 x 13 1575792
(12) maxpool 512 x 13 x 13 512 x 13 x 13 1575792
(13) conv-bn-leaky 512 x 13 x 13 1024 x 13 x 13 6298480
(14) conv-bn-leaky 1024 x 13 x 13 512 x 13 x 13 11019120
(15) conv-linear 512 x 13 x 13 95 x 13 x 13 11067855
(16) region 95 x 13 x 13 95 x 13 x 13 11067855
weights.size():11067854
weightPtr:11067855
Number of unused weights left : 18446744073709551615
trt-yolo-app: yolo.cpp:399: void Yolo::createYOLOEngine(int, std::__cxx11::string, std::__cxx11::string, std::__cxx11::string, nvinfer1::DataType, Int8EntropyCalibrator*): Assertion `0' failed.
Aborted (core dumped)
tihis model can run in the repo,but there is bug in trt-yolo-app.
This question has been solved by retraining a new model.
Thanks.
Most helpful comment
This question has been solved by retraining a new model.
Thanks.