[infer_keypoint]fix for visualization part

[jacinto-ai/caffe-jacinto.git] / src / caffe / layers / detection_output_layer.cu

diff --git a/src/caffe/layers/detection_output_layer.cu b/src/caffe/layers/detection_output_layer.cu
index 1b02185e07f3434e1055acdd2e9dbd4b9b6c16e6..67b8ada66516d5a91e0ed9e0a21beeb6c64a6f6a 100644 (file)
--- a/src/caffe/layers/detection_output_layer.cu
+++ b/src/caffe/layers/detection_output_layer.cu
@@ -297,7 +297,7 @@ void DetectionOutputLayer<Ftype, Btype>::Forward_gpu(
     this->data_transformer_->TransformInv(bottom[3], &cv_imgs);
     vector<cv::Scalar> colors = GetColors(label_to_display_name_.size());
     VisualizeBBox<Ftype>(cv_imgs, top[0], visualize_threshold_, colors,
-        label_to_display_name_, save_file_);
+        label_to_display_name_, save_file_, num_keypoint_);
   }
 }