增加end2end方式计算位姿

vision_detect/vision_detect/VisionDetect/net/knn/src/knn.h

@@ -14,7 +14,7 @@
 int knn(at::Tensor& ref, at::Tensor& query, at::Tensor& idx)
 {
 
-    // TODO check dimensions
+    // NOTE: check dimensions
     long batch, ref_nb, query_nb, dim, k;
     batch = ref.size(0);
     dim = ref.size(1);
@@ -31,7 +31,7 @@ int knn(at::Tensor& ref, at::Tensor& query, at::Tensor& idx)
 
   if (ref.type().is_cuda()) {
 #ifdef WITH_CUDA
-    // TODO raise error if not compiled with CUDA
+    // NOTE: raise error if not compiled with CUDA
 //    float *dist_dev = (float*)THCudaMalloc(state, ref_nb * query_nb * sizeof(float));
     float *dist_dev = nullptr;
     cudaError_t err = cudaMalloc((void**)&dist_dev, ref_nb * query_nb * sizeof(float));