一种基于TX5368A平台的图像分割方法及设备

本申请公开了一种基于TX5368A平台的图像分割方法及设备，涉及图像分割技术领域，该方法包括：构建图像分割网络；图像分割网络是基于YOLOv5s‑seg 7.0网络架构搭建的；以Pytorch为训练框架，利用训练集对图像分割网络进行训练，得到最优Pytorch模型；将最优Pytorch模型转换为onnx模型；对onnx模型进行量化处理，将量化处理后的onnx模型部署在安装有图像获取装置的TX5368A平台上，本申请通过构建并训练图像分割网络，将图像分割网络部署于TX5368A平台，可对图像分割算法进行轻量化处理，从而降低部署平台的算力要求。