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
, Dot
AveragePooling1D/2D/3D
, GlobalAveragePooling1D/2D/3D
TimeDistributed
Conv1D/2D
, SeparableConv2D
, DepthwiseConv2D
Cropping1D/2D/3D
, ZeroPadding1D/2D/3D
, CenterCrop
BatchNormalization
, Dense
, Flatten
, Normalization
Dropout
, AlphaDropout
, GaussianDropout
, GaussianNoise
SpatialDropout1D
, SpatialDropout2D
, SpatialDropout3D
ActivityRegularization
, LayerNormalization
, UnitNormalization
RandomContrast
, RandomFlip
, RandomHeight
RandomRotation
, RandomTranslation
, RandomWidth
, RandomZoom
MaxPooling1D/2D/3D
, GlobalMaxPooling1D/2D/3D
ELU
, LeakyReLU
, ReLU
, SeLU
, PReLU
Sigmoid
, Softmax
, Softplus
, Tanh
Exponential
, GELU
, Softsign
, Rescaling
UpSampling1D/2D
, Resizing
Reshape
, Permute
, RepeatVector
Embedding
, CategoryEncoding
Attention
, AdditiveAttention
, MultiHeadAttention
load_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)