eblite_smart_car

The edgeboard lite code on python api for the smart car.

目录结构

库

model.py -- 模型加载

tool.py -- 一些工具

devices.py -- 智能车设备

preprocess.py -- 数据预处理

Demo

auto_driver(_video).py -- 自动行车程序(行车视频)

auto_tracking(_video).py -- 原地追踪程序(行车视频)

data_collection.py -- 遥控数据采集程序(Demo版)

模型（车道线模型效果不好，仅供测试）

ssd_lite -- 检测模型（ssd_lite）

car_line -- 车道线模型

API

所有API均在代码中有详细的使用注释，方便调用和二次修改使用

快速使用

# 导入需要的包
from model import pm_model
from devices import car_devices
from preprocess import preprocess_det

# 加载模型
ssd_lite = pm_model(data_shape=(1, 3, 128, 128), model_flie='./ssd_lite/model', param_file='./ssd_lite/params')

# 初始化小车设备
car = car_devices()

# 读取小车摄像头图像
frame = car.read_frame()

# 预处理数据
img = preprocess_det(frame, (128, 128))

# 模型预测
result = ssd_lite.predict(img)

# 打印结果
print(result)

如何部署自己的模型

1. 使用paddlepaddle训练自己的模型

2. 导出推理模型

3. 使用model模块中pm_model或cxx_model这两个api加载模型

4. 根据自己的模型写好数据的预处理代码

5. 调用model类的predict函数进行预测