Add YOLOv7 support
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user