enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.19k stars 315 forks source link

batch checking is missing in the yolo.cpp #153

Open DiamondSheep opened 2 years ago

DiamondSheep commented 2 years ago

In the function of yolo.cpp void Yolo::parse_cfg_blocks_v5(const std::vector<std::map<std::string, std::string>> &vec_block_) , the "batch" checking is missing, assert((block.find("batch") != block.end()) && "Missing 'batch' param in network cfg"); should be added at 1179 line.