0
点赞
收藏
分享

微信扫一扫

20 基于STM32的温度、电流、电压检测proteus仿真系统(OLED、DHT11、继电器、电机)

勇敢乌龟 2024-10-09 阅读 25

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

安装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格式的文件

举报

相关推荐

0 条评论