CMakeLists.txt 3.3 KB
Newer Older
B
baiyf 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
set(LOCAL_DETECTION_LIBS)

function(detection_library TARGET_NAME)
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    set(options "")
    set(common_deps op_registry)
    set(pybind_flag 0)
    cmake_parse_arguments(detection_library "${options}" "${oneValueArgs}"
            "${multiValueArgs}" ${ARGN})
    op_library(${TARGET_NAME} SRCS ${detection_library_SRCS} DEPS ${common_deps} ${detection_library_DEPS})
    set(LOCAL_DETECTION_LIBS
            ${TARGET_NAME}
            ${LOCAL_DETECTION_LIBS}
        PARENT_SCOPE)
endfunction()

detection_library(bipartite_match_op SRCS bipartite_match_op.cc)
detection_library(box_coder_op SRCS box_coder_op.cc box_coder_op.cu)
detection_library(iou_similarity_op SRCS iou_similarity_op.cc
iou_similarity_op.cu)
detection_library(mine_hard_examples_op SRCS mine_hard_examples_op.cc)
detection_library(prior_box_op SRCS prior_box_op.cc prior_box_op.cu)
24
detection_library(density_prior_box_op SRCS density_prior_box_op.cc density_prior_box_op.cu)
25 26
detection_library(anchor_generator_op SRCS anchor_generator_op.cc
anchor_generator_op.cu)
B
baiyf 已提交
27 28
detection_library(target_assign_op SRCS target_assign_op.cc
target_assign_op.cu)
29
detection_library(polygon_box_transform_op SRCS polygon_box_transform_op.cc
Y
Yuan Gao 已提交
30 31
polygon_box_transform_op.cu)
detection_library(rpn_target_assign_op SRCS rpn_target_assign_op.cc)
32
detection_library(generate_proposal_labels_op SRCS generate_proposal_labels_op.cc)
33 34
detection_library(multiclass_nms_op SRCS multiclass_nms_op.cc DEPS gpc)
detection_library(locality_aware_nms_op SRCS locality_aware_nms_op.cc DEPS gpc)
Y
Yang Zhang 已提交
35
detection_library(matrix_nms_op SRCS matrix_nms_op.cc DEPS gpc)
36
detection_library(box_clip_op SRCS box_clip_op.cc box_clip_op.cu)
37
detection_library(yolov3_loss_op SRCS yolov3_loss_op.cc)
D
dengkaipeng 已提交
38
detection_library(yolo_box_op SRCS yolo_box_op.cc yolo_box_op.cu)
39
detection_library(box_decoder_and_assign_op SRCS box_decoder_and_assign_op.cc box_decoder_and_assign_op.cu)
40
detection_library(sigmoid_focal_loss_op SRCS sigmoid_focal_loss_op.cc sigmoid_focal_loss_op.cu)
41
detection_library(retinanet_detection_output_op SRCS retinanet_detection_output_op.cc)
42 43

if(WITH_GPU)
44 45 46 47 48 49 50
  set(TMPDEPS memory)
  if (${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0)
      set(TMPDEPS memory cub)
  endif()
  detection_library(generate_proposals_op SRCS generate_proposals_op.cc generate_proposals_op.cu DEPS ${TMPDEPS})
  detection_library(distribute_fpn_proposals_op SRCS distribute_fpn_proposals_op.cc distribute_fpn_proposals_op.cu DEPS ${TMPDEPS})
  detection_library(collect_fpn_proposals_op SRCS collect_fpn_proposals_op.cc collect_fpn_proposals_op.cu DEPS ${TMPDEPS})
51 52
else()
  detection_library(generate_proposals_op SRCS generate_proposals_op.cc)
J
jerrywgz 已提交
53
  detection_library(distribute_fpn_proposals_op SRCS distribute_fpn_proposals_op.cc)
54
  detection_library(collect_fpn_proposals_op SRCS collect_fpn_proposals_op.cc)
55 56
endif()

W
whs 已提交
57
detection_library(roi_perspective_transform_op SRCS roi_perspective_transform_op.cc roi_perspective_transform_op.cu)
58
#Export local libraries to parent
W
Wu Yi 已提交
59 60
# set(DETECTION_LIBRARY ${LOCAL_DETECTION_LIBS} PARENT_SCOPE)

61 62
cc_library(mask_util SRCS mask_util.cc DEPS memory)
cc_test(mask_util_test SRCS mask_util_test.cc DEPS memory mask_util)
63
cc_library(gpc SRCS gpc.cc DEPS op_registry)
64
detection_library(generate_mask_labels_op SRCS generate_mask_labels_op.cc DEPS mask_util)