pointnet train函數第十六至第二十句

原創 guyuezunting 2020-07-02 19:03 
config = tf.ConfigProto()
        config.gpu_options.allow_growth = True
        config.allow_soft_placement = True
        config.log_device_placement = False
        sess = tf.Session(config=config)

其十六~十九，屬於即將初始化的session的config設置

十六句初始化一個config，後面三句分別爲

動態申請顯存爲true

設置自動選擇運行設備爲true

設置記錄，operation以及tensor在哪個設備上運行爲false

二十句則爲初始化session配置爲config

發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

open3d numpy array 轉換爲point cloud

在三維點雲處理過程中經常會自己造點點雲數據進行展示，但是一般情況下都是用numpy array甚至是list把數據先存起來，但是open3d裏面的 visualization.draw_geometries（[pcd]）裏面pcd是po

guyuezunting 2020-07-08 09:00:51

pointnet pytorch-unet網絡學習一

爲了用pytorch實現pointnet，需要先學習pytorch的一個網絡，然後再進行實現 關於pytorch的安裝，個人建議初學者安裝cpu的最好，因爲gpu的pytorch在windows下特別難以安裝成功 因爲gpu目前實在版本混

guyuezunting 2020-07-02 19:03:43

pointnet學習（八）tf_util.conv2d

pointnet學習（七）中input_transform_net函數，通過條用tf_util.conv2d構建net。 tf_util.conv2d(input_image, 64, [1,3],

guyuezunting 2020-07-02 19:03:43

pintnet tf_util.dropout

net = tf_util.dropout(net, keep_prob=0.7, is_training=is_training, scope='dp1') 具體實現如下 def drop

guyuezunting 2020-07-02 19:03:43

pointnet provider.loadDataFile

首先看具體實現 def load_h5(h5_filename): f = h5py.File(h5_filename) data = f['data'][:] label = f['label'][:]

guyuezunting 2020-07-02 19:03:43

pointnet tf_util.max_pool2d

net = tf_util.max_pool2d(net, [num_point,1], padding='VALID', scope='tmaxpool') 看下具體實現 d

guyuezunting 2020-07-02 19:03:43

pointnet學習（六）train函數第七第八句

   第七句 tf.summary.scalar('bn_decay', bn_decay) 查看pointnet學習（二）tf.summary.scalar含義  第八句 pred, end_points = MODEL.get

guyuezunting 2020-07-02 19:03:43

pointnet train函數第二十七句 for epoch in range(MAX_EPOCH):

for epoch in range(MAX_EPOCH): log_string('**** EPOCH %03d ****' % (epoch)) sys.stdout

guyuezunting 2020-07-02 19:03:43

pointnet train 第十句correct = tf.equal(tf.argmax(pred, 1), tf.to_int64(labels_pl))

correct = tf.equal(tf.argmax(pred, 1), tf.to_int64(labels_pl)) 首先看tf.argmax(pred,1)因爲pred,是32，40的tensor，40那一維是40個模型中對應

guyuezunting 2020-07-02 19:03:43

pointnet train函數二十一至二十三句

merged = tf.summary.merge_all() train_writer = tf.summary.FileWriter(os.path.join(LOG_DIR, 'train'),

guyuezunting 2020-07-02 19:03:43

pointnet （九）train函數，第九句

loss = MODEL.get_loss(pred, labels_pl, end_points) 第七句，第八句是獲得net以及transformnet，最終得到的值需要與label一起進行計算pointnet學習（六）train函

guyuezunting 2020-07-02 19:03:43

pointnet tf_util.fully_connected

net = tf_util.fully_connected(net, 512, bn=True, is_training=is_training, scope='tfc

guyuezunting 2020-07-02 19:03:43

pointnet train函數第二十五句 sess.run(init, {is_training_pl: True})

sess.run(init, {is_training_pl: True}) 運行第二十四句的初始化op，同時初始化is_training_pl這個bool類型tensor初始值爲true。此tensor在運行模型過程中多次用到

guyuezunting 2020-07-02 19:03:43

pointnet train函數第二十六句 ops

ops = {'pointclouds_pl': pointclouds_pl, 'labels_pl': labels_pl, 'is_training_pl': is_tr

guyuezunting 2020-07-02 19:03:43

pointnet input_transform_net2

先看工程中的調用 with tf.variable_scope('transform_net2') as sc: transform = feature_transform_net(net, is_trainin

guyuezunting 2020-07-02 19:03:43