使用自己的数据集训练模型进行图像分类
安装Model Maker 软件包
sudo apt -y install libportaudio2pip install -q tflite-model-maker
准备数据集
文件夹下各类别子文件夹，图片文件
导入包
import os
import numpy as np
import 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 DataLoader
import 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格式的文件
