Deepstream 5.0-Custom Yolo Model num classes mismatch

需要对文件进行修改

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

你可能感兴趣的:(Jetson,Nano)