DeepStream Yolo
1.0.0
Nvidia Deepstream SDK 7.1 / 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 Configuração para modelos YOLO
git clone https://github.com/marcoslucianops/DeepStream-Yolo.git
cd DeepStream-Yolo
cfg e weights de DarkNet Repo até a pasta Deepstream-Yolo 3.1. Defina o CUDA_VER de acordo com sua versão Deepstream
export CUDA_VER=XY.Z
Plataforma x86
DeepStream 7.1 = 12.6
DeepStream 7.0 / 6.4 = 12.2
DeepStream 6.3 = 12.1
DeepStream 6.2 = 11.8
DeepStream 6.1.1 = 11.7
DeepStream 6.1 = 11.6
DeepStream 6.0.1 / 6.0 = 11.4
DeepStream 5.1 = 11.1
Plataforma Jetson
DeepStream 7.1 = 12.6
DeepStream 7.0 / 6.4 = 12.2
DeepStream 6.3 / 6.2 / 6.1.1 / 6.1 = 11.4
DeepStream 6.0.1 / 6.0 / 5.1 = 10.2
3.2. Faça o Lib
make -C nvdsinfer_custom_impl_Yolo clean && make -C nvdsinfer_custom_impl_Yolo
config_infer_primary.txt de acordo com o seu modelo (exemplo para yolov4) [property]
...
custom-network-config=yolov4.cfg
model-file=yolov4.weights
...
Nota : Para modelos DarkNet , por padrão, o size em lote dinâmico está definido. Para usar o tamanho de lote estático, descommentando a linha
...
force-implicit-batch-dim=1
...
deepstream-app -c deepstream_app_config.txt
NOTA : O arquivo do motor Tensorrt pode levar muito tempo para gerar (às vezes mais de 10 minutos).
Nota : Se você deseja usar modelos Yolov2 ou Yolov2, altere o arquivo deepstream_app_config.txt antes de executá-lo
...
[primary-gie]
...
config-file=config_infer_primary_yoloV2.txt
...
Plataforma x86
nvcr.io/nvidia/deepstream:7.1-gc-triton-devel
nvcr.io/nvidia/deepstream:7.1-triton-multiarch
Plataforma Jetson
nvcr.io/nvidia/deepstream:7.1-triton-multiarch
Nota : Para compilar o nvdsinfer_custom_impl_Yolo , você precisa instalar o G ++ dentro do contêiner
apt-get install build-essential
NOTA : Com o Deepstream 7.1, os contêineres do Docker não embalam bibliotecas necessárias para determinadas operações multimídia, como análise de dados de áudio, decodificação da CPU e codificação da CPU. Essa alteração pode afetar o processamento de determinados fluxos/arquivos de vídeo, como o MP4, que incluem faixa de áudio. Execute o script abaixo dentro das imagens do Docker para instalar pacotes adicionais que podem ser necessários para usar todos os recursos do DeepStreamsdk:
/opt/nvidia/deepstream/deepstream/user_additional_install.sh
Para alterar os valores nms-iou-threshold , pre-cluster-threshold e topk , modifique o arquivo config_infer
[class-attrs-all]
nms-iou-threshold=0.45
pre-cluster-threshold=0.25
topk=300
Nota : Certifique-se de definir cluster-mode=2 no arquivo config_infer.
Às vezes, ao executar aplicativos GSTreamer ou aplicativos de amostra, o usuário pode encontrar erro: GLib (gthread-posix.c): Unexpected error from C library during 'pthread_setspecific': Invalid argument. Aborting. . O problema é causado devido a um bug na versão glib 2.0-2.72 , que vem com o Ubuntu 22.04 por padrão. O problema é abordado no glib 2.76 e sua instalação é necessária para corrigir o problema (https://github.com/gnome/glib/tree/2.76.6).
Migrar glib para uma versão mais recente
pip3 install meson
pip3 install ninja
Nota : é recomendável usar o Python VirtualEnv.
git clone https://github.com/GNOME/glib.git
cd glib
git checkout 2.76.6
meson build --prefix=/usr
ninja -C build/
cd build/
ninja install
Verifique e confirme a versão Glib recém -instalada:
pkg-config --modversion glib-2.0
Às vezes, com os fluxos RTSP, o aplicativo fica preso ao atingir o EOS. Isso ocorre devido a um problema no componente rtpjitterbuffer. Para corrigir esse problema, um script recebeu os detalhes necessários para atualizar a biblioteca GSTRTPManager.
/opt/nvidia/deepstream/deepstream/update_rtpmanager.sh
Você pode obter metadados do DeepStream usando Python e C/C ++. Para C/C ++, você pode editar os códigos deepstream-app ou deepstream-test . Para Python, você pode instalar e editar Deepstream_python_apps.
Basicamente, você precisa manipular o NvDsObjectMeta (python / c / c ++) and NvDsFrameMeta (python / c / c ++) para obter o rótulo, a posição etc. de Bboxes.
Meus projetos: https://www.youtube.com/marcoslucianotv
