深度学习——基于点云的车辆检测三维全卷积网络

 '''

    
 Baidu Inc. 
    
   4. Ref: 
    
 3D Fully Convolutional Network for Vehicle Detection in Point Cloud
    
   7. Author: HSW 
    
 Date: 2018-05-02 
    
 '''
    
  
    
  
    
 import sys
    
 import numpy as np 
    
 import tensorflow as tf 
    
 from prepare_data2 import * 
    
 from baidu_cnn_3d import * 
    
  
    
 KITTI_TRAIN_DATA_CNT = 7481
    
 KITTI_TEST_DATA_CNT  = 7518
    
  
    
  
    
 # create 3D-CNN Model
    
 def create_graph(sess, modelType = 0, voxel_shape = (400, 400, 20),  activation=tf.nn.relu, is