FengD / npy_2_pb

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

npy_2_pb

Description

This file is used to transform the npy file to pb file. And I create this file especially for the Squeezeseg project. The author of SqueezeSeg transform the kitti data to lidar_2d as he mentioned in his paper. This dataset contains LiDAR point-cloud projected to a 2D spherical surface. Refer to our paper for details of the data conversion procedure.

And I create the script in the project to transform his .npy file type to the .pb file type which could be used in tensorflow.

Dependencies

  • tensorflow
  • numpy
  • glob

Tips

For the line 22, if you meet the error below:

AttributeError: module 'tensorflow' has no attribute 'reset_default_graph'

Flow this link

Just change

import tensorflow as tf

to

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

About


Languages

Language:Python 100.0%