Custom gst-nvinfer (DEMO)

This is a custom gst-nvinfer plugin to do some preprocess.

Requirements

export CUDA_VER=11.6

make
make install

NOTE: To compile the sources, run make with "sudo" or root permission.

first to set primary gie's output-tensor-meta to true. For example, in retinaface config file:

output-tensor-meta=1

use kyeword alignment and user-meta in next gie-config file. For example, in arcface config file:

alignment=1
user-meta=1

Now only retinaface and arcface(operate on retinaface)

the custom gst-nvinfer has two new properties: alignment and user-meta.

Once these properties are true, there would be a preprocess on every croped object.

It first receives the output tensor meta, to extract the 5 landmarks of a face and do NMS. By similarTransform, a matrix M will be generated.

then use the croped object and M to do alignment: cv::warpPerspective.

finally, cover the original surface with the output of alignment.

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
Makefile		Makefile
README.md		README.md
README_nv		README_nv
arcface_preprocess.cpp		arcface_preprocess.cpp
arcface_preprocess.h		arcface_preprocess.h
gstnvinfer.cpp		gstnvinfer.cpp
gstnvinfer.h		gstnvinfer.h
gstnvinfer.o		gstnvinfer.o
gstnvinfer_allocator.cpp		gstnvinfer_allocator.cpp
gstnvinfer_allocator.h		gstnvinfer_allocator.h
gstnvinfer_allocator.o		gstnvinfer_allocator.o
gstnvinfer_impl.cpp		gstnvinfer_impl.cpp
gstnvinfer_impl.h		gstnvinfer_impl.h
gstnvinfer_impl.o		gstnvinfer_impl.o
gstnvinfer_meta_utils.cpp		gstnvinfer_meta_utils.cpp
gstnvinfer_meta_utils.h		gstnvinfer_meta_utils.h
gstnvinfer_meta_utils.o		gstnvinfer_meta_utils.o
gstnvinfer_property_parser.cpp		gstnvinfer_property_parser.cpp
gstnvinfer_property_parser.h		gstnvinfer_property_parser.h
gstnvinfer_property_parser.o		gstnvinfer_property_parser.o
gstnvinfer_yaml_parser.cpp		gstnvinfer_yaml_parser.cpp
gstnvinfer_yaml_parser.h		gstnvinfer_yaml_parser.h
gstnvinfer_yaml_parser.o		gstnvinfer_yaml_parser.o
libnvdsgst_infer.so		libnvdsgst_infer.so