点云ESF特征分类(神经网络)

通过对点云进行ESF全局特征提取,训练神经网络分类器,对点云进行分类。ESF特征提取见上述链接。
基于tensorflow2
python代码如下:

import tensorflow as tf
import numpy as np
dataset =tf.data.Dataset

a3 = np.loadtxt('')
train1 = np.array(a3)
label1 = np.loadtxt('')
label1 = tf.one_hot(label1,depth=3)

a4 = np.loadtxt('')
text = np.array(a4)
label_text = np.loadtxt('')
label_text = tf.one_hot(label_text,depth=3)

input_data = tf.keras.Input(shape=(640,),name='input_data')
##out=tf.keras.layers.Flatten()(input_data)

##out=tf.keras.layers.Dense(512,activation='relu',name='dense_1')(input_data)

##ou=tf.keras.layers.Dense(256,activation='relu',name='dense_2')(out)

out1=tf.keras.layers.Dense(128,activation='relu',name='dense_3')(input_data)

out2=tf.keras.layers.Dense(128,activation='relu',name='dense_4')(out1)

out3=tf.keras.layers.Dense(32,activation='relu',name='dense_5')(out2)

pre=tf.keras.layers.Dense(3,activation='softmax',name='predictions')(out3)

model=tf.keras.Model(inputs=input_data,outputs=pre)

print(model.summary())

model.compile(optimizer=tf.keras.optimizers.Adam(1e-4),
loss='categorical_crossentropy',metrics=['accuracy'])

tensorboard = tf.keras.callbacks.TensorBoard(histogram_freq=1)
model.fit(x=train1,y=label1,epochs=500,batch_size=1,verbose=2,callbacks=[tensorboard])
model.evaluate(x=text,y=label_text)
y_pred = model.predict(text,batch_size = 1)
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值