Gunakan model Keras di C++ dengan mudah
Apakah Anda ingin membuat/melatih model menggunakan Keras/Python? Dan apakah Anda ingin menjalankan prediksi (forward pass) pada model Anda di C++ tanpa menautkan aplikasi Anda ke TensorFlow? Maka hemat dalam-dalam adalah pilihan yang tepat untuk Anda.
sangat hemat
model.predict ) tidak hanya untuk model sekuensial tetapi juga untuk grafik komputasi dengan topologi yang lebih kompleks, dibuat dengan API fungsional.Add , Concatenate , Subtract , Multiply , Average , Maximum , Minimum , DotAveragePooling1D/2D/3D , GlobalAveragePooling1D/2D/3DTimeDistributedConv1D/2D , SeparableConv2D , DepthwiseConv2DCropping1D/2D/3D , ZeroPadding1D/2D/3D , CenterCropBatchNormalization , Dense , Flatten , NormalizationDropout , AlphaDropout , GaussianDropout , GaussianNoiseSpatialDropout1D , SpatialDropout2D , SpatialDropout3DActivityRegularization , LayerNormalization , UnitNormalizationRandomContrast , RandomFlip , RandomHeightRandomRotation , RandomTranslation , RandomWidth , RandomZoomMaxPooling1D/2D/3D , GlobalMaxPooling1D/2D/3DELU , LeakyReLU , ReLU , SeLU , PReLUSigmoid , Softmax , Softplus , TanhExponential , GELU , Softsign , RescalingUpSampling1D/2D , ResizingReshape , Permute , RepeatVectorEmbedding , CategoryEncodingAttention , AdditiveAttention , MultiHeadAttentionload_model ) Conv2DTranspose (mengapa), Lambda (mengapa), Conv3D , ConvLSTM1D , ConvLSTM2D , Discretization , GRUCell , Hashing , IntegerLookup , LocallyConnected1D , LocallyConnected2D , LSTMCell , Masking , RepeatVector , RNN , SimpleRNN , SimpleRNNCell , StackedRNNCells , StringLookup , TextVectorization , Bidirectional , GRU , LSTM , CuDNNGRU , CuDNNLSTM , ThresholdedReLU , Upsampling3D , model temporal
Gunakan Keras/Python untuk membangun ( model.compile(...) ), melatih ( model.fit(...) ) dan menguji ( model.evaluate(...) ) model Anda seperti biasa. Kemudian simpan ke satu file menggunakan model.save('....keras') . image_data_format dalam model Anda channels_last , yang merupakan default saat menggunakan backend TensorFlow. Model yang dibuat dengan image_data_format dan backend lain yang berbeda tidak didukung.
Sekarang konversikan ke format file yang hemat dengan keras_export/convert_model.py
Terakhir muat dalam C++ ( fdeep::load_model(...) ) dan gunakan model.predict(...) untuk menjalankan forward pass dengan data Anda.
Contoh minimal berikut menunjukkan alur kerja lengkap:
# create_model.py
import numpy as np
from tensorflow . keras . layers import Input , Dense
from tensorflow . keras . models import Model
inputs = Input ( shape = ( 4 ,))
x = Dense ( 5 , activation = 'relu' )( inputs )
predictions = Dense ( 3 , activation = 'softmax' )( x )
model = Model ( inputs = inputs , outputs = predictions )
model . compile ( loss = 'categorical_crossentropy' , optimizer = 'nadam' )
model . fit (
np . asarray ([[ 1 , 2 , 3 , 4 ], [ 2 , 3 , 4 , 5 ]]),
np . asarray ([[ 1 , 0 , 0 ], [ 0 , 0 , 1 ]]), epochs = 10 )
model . save ( 'keras_model.keras' )python3 keras_export/convert_model.py keras_model.keras fdeep_model.json // main.cpp
# include < fdeep/fdeep.hpp >
int main ()
{
const auto model = fdeep::load_model ( " fdeep_model.json " );
const auto result = model. predict (
{ fdeep::tensor ( fdeep::tensor_shape ( static_cast <std:: size_t >( 4 )),
std::vector< float >{ 1 , 2 , 3 , 4 })});
std::cout << fdeep::show_tensors (result) << std::endl;
} Saat menggunakan convert_model.py kasus uji (input dan nilai output terkait) dihasilkan secara otomatis dan disimpan bersama dengan model Anda. fdeep::load_model menjalankan tes ini untuk memastikan hasil forward pass di frugally-deep sama dengan di Keras.
Untuk contoh integrasi lainnya, silakan lihat FAQ.
Panduan untuk berbagai cara menginstal secara hemat dapat ditemukan di INSTALL.md .
Lihat FAQ.md
API perpustakaan ini mungkin masih berubah di masa mendatang. Jika Anda mempunyai saran, menemukan kesalahan, atau ingin memberikan masukan/kritik umum, saya ingin mendengar pendapat Anda. Tentu saja, kontribusinya juga sangat diharapkan.
Didistribusikan di bawah Lisensi MIT. (Lihat file yang menyertai LICENSE atau di https://opensource.org/licenses/MIT)