使用自己的数据集训练模型进行图像分类

安装Model Maker 软件包

sudo apt -y install libportaudio2pip install -q tflite-model-maker

准备数据集

文件夹下各类别子文件夹，图片文件

导入包

import osimport numpy as npimport tensorflow as tf
assert tf.__version__.startswith('2')from tflite_model_maker import model_spec
from tflite_model_maker import image_classifier
from tflite_model_maker.config import ExportFormat
from tflite_model_maker.config import QuantizationConfig
from tflite_model_maker.image_classifier import DataLoaderimport matplotlib.pyplot as plt

运行

data = DataLoader.from_folder(image_path)
train_data, test_data = data.split(0.9)

model = image_classifier.create(train_data)

loss, accuracy = model.evaluate(test_data)

model.export(export_dir='.')

导出模型即为tflite格式的文件