Caffe2 - Memverifikasi Akses ke Model yang Dilatih

Sebelum Anda belajar menggunakan model yang telah dilatih sebelumnya dalam aplikasi Python Anda, mari kita verifikasi terlebih dahulu bahwa model tersebut diinstal pada mesin Anda dan dapat diakses melalui kode Python.

Saat Anda menginstal Caffe2, model terlatih disalin di folder instalasi. Pada mesin dengan penginstalan Anaconda, model ini tersedia di folder berikut.

anaconda3/lib/python3.7/site-packages/caffe2/python/models

Lihat folder instalasi pada mesin Anda untuk mengetahui keberadaan model-model ini. Anda dapat mencoba memuat model ini dari folder instalasi dengan skrip singkat Python berikut -

CAFFE_MODELS = os.path.expanduser("/anaconda3/lib/python3.7/site-packages/caffe2/python/models")
INIT_NET = os.path.join(CAFFE_MODELS, 'squeezenet', 'init_net.pb')
PREDICT_NET = os.path.join(CAFFE_MODELS, 'squeezenet', 'predict_net.pb')
print(INIT_NET)
print(PREDICT_NET)

Ketika skrip berhasil dijalankan, Anda akan melihat output berikut -

/anaconda3/lib/python3.7/site-packages/caffe2/python/models/squeezenet/init_net.pb
/anaconda3/lib/python3.7/site-packages/caffe2/python/models/squeezenet/predict_net.pb

Ini menegaskan bahwa squeezenet modul diinstal pada mesin Anda dan dapat diakses oleh kode Anda.

Sekarang, Anda siap untuk menulis kode Python Anda sendiri untuk klasifikasi gambar menggunakan Caffe2 squeezenet modul terlatih.