需要对文件进行修改
deepstream/deepstream-5.0/sources/objectDetector_Yolo/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp
static const int NUM_CLASSES_YOLO = 3; #修改为训练的类别数量
#重新编译
export CUDA_VER=10.2
make
deepstream/deepstream-5.0/sources/objectDetector_Yolo/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp
static const int NUM_CLASSES_YOLO = 3; #修改为训练的类别数量
#重新编译
export CUDA_VER=10.2
make