예제 #1
0
파일: network.py 프로젝트: zuwuwang/PoseCNN
 def hough_voting_gpu(self, input, is_train, threshold, skip_pixels, name):
     return hough_voting_gpu_op.hough_voting_gpu(input[0],
                                                 input[1],
                                                 input[2],
                                                 input[3],
                                                 input[4],
                                                 is_train,
                                                 threshold,
                                                 skip_pixels,
                                                 name=name)
예제 #2
0
 def hough_voting_gpu(self, input, kernel_size, threshold, per_threshold,
                      skip_pixels, name):
     return hough_voting_gpu_op.hough_voting_gpu(input[0], input[1], input[2], input[3], input[4], input[5], input[6], kernel_size, \
                threshold, per_threshold, skip_pixels, name=name)