RK3568笔记二十二：基于TACO的垃圾检测和识别

若该文为原创文章，转载请注明原文出处。

基于TACO数据集，使用YOLOv8分割模型进行垃圾检测和识别，并在ATK-RK3568上部署运行。

一、环境

1、测试训练环境：AutoDL.

2、平台：rk3568

3、开发板: ATK-RK3568正点原子板子

4、环境：buildroot

5、虚拟机：正点原子提供的ubuntu 20

二、测试

个人电脑没有GPU，在AutoDL租了服务器，配置如下：将在上面测试并训练。

PyTorch  1.8.1
Python  3.8(ubuntu18.04) 

CUDA 11.0

三、环境安装

1、使用conda创建虚拟环境

conda create -n yolov8 python=3.8
conda activate yolov8

2、 安装YOLOv8

# 拉取代码
git clone https://github.com/ultralytics/ultralytics
cd ultralytics

pip install -e .

三、下载TACO数据集