|
|
|
@ -31,6 +31,7 @@ detection_library(polygon_box_transform_op SRCS polygon_box_transform_op.cc
|
|
|
|
|
polygon_box_transform_op.cu)
|
|
|
|
|
detection_library(rpn_target_assign_op SRCS rpn_target_assign_op.cc)
|
|
|
|
|
detection_library(generate_proposal_labels_op SRCS generate_proposal_labels_op.cc)
|
|
|
|
|
detection_library(yolov3_loss_op SRCS yolov3_loss_op.cc)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
detection_library(generate_proposals_op SRCS generate_proposals_op.cc generate_proposals_op.cu DEPS memory cub)
|
|
|
|
|