Add YOLOv7 support

This commit is contained in:
Marcos Luciano
2022-08-12 16:33:26 -03:00
parent 23547b19b2
commit 80d08990a0
10 changed files with 586 additions and 47 deletions

View File

@@ -49,37 +49,16 @@ LIBS+= -lnvinfer_plugin -lnvinfer -lnvparsers -L/usr/local/cuda-$(CUDA_VER)/lib6
LFLAGS:= -shared -Wl,--start-group $(LIBS) -Wl,--end-group
INCS:= $(wildcard *.h)
SRCFILES:= nvdsinfer_yolo_engine.cpp \
nvdsparsebbox_Yolo.cpp \
yoloPlugins.cpp \
layers/convolutional_layer.cpp \
layers/batchnorm_layer.cpp \
layers/implicit_layer.cpp \
layers/channels_layer.cpp \
layers/shortcut_layer.cpp \
layers/route_layer.cpp \
layers/upsample_layer.cpp \
layers/pooling_layer.cpp \
layers/activation_layer.cpp \
layers/reorg_layer.cpp \
layers/reduce_layer.cpp \
layers/shuffle_layer.cpp \
layers/softmax_layer.cpp \
layers/cls_layer.cpp \
layers/reg_layer.cpp \
utils.cpp \
yolo.cpp \
yoloForward.cu \
yoloForward_v2.cu \
yoloForward_nc.cu \
yoloForward_r.cu \
yoloForward_e.cu \
sortDetections.cu
SRCFILES:= $(filter-out calibrator.cpp, $(wildcard *.cpp))
ifeq ($(OPENCV), 1)
SRCFILES+= calibrator.cpp
endif
SRCFILES+= $(wildcard layers/*.cpp)
SRCFILES+= $(wildcard *.cu)
TARGET_LIB:= libnvdsinfer_custom_impl_Yolo.so
TARGET_OBJS:= $(SRCFILES:.cpp=.o)