尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
水面漂浮垃圾数据集进行目标检测 识别11类河道漂浮数据集中的玻璃垃圾 泡沫塑料碎片垃圾 金属垃圾 PET塑料瓶垃圾 其他塑料垃圾
使用YOLOv8对水面漂浮垃圾数据集进行目标检测 识别11类河道漂浮数据集中的玻璃垃圾 泡沫塑料碎片垃圾 金属垃圾 PET塑料瓶垃圾 其他塑料垃圾文章目录一、环境准备安装CUDA驱动安装Anaconda并创建虚拟环境安装依赖项二、数据集组织三、使用YOLOv8训练模型加载预训练模型训练模型四、推理代码五、评估代码以下文字及代码仅供参考学习使用。数据集描述水面漂浮垃圾检测数据集11类Glass–玻璃垃圾Styrofoam_Piece --泡沫塑料碎片垃圾Plastic_Buoy_China --塑料浮标垃圾中国产Metal --金属垃圾Net --废弃渔网Rope --废弃绳索Plastic_Buoy --塑料浮标垃圾PET_Bottle --PET塑料瓶垃圾Styrofoam_Box --泡沫箱垃圾Plastic_ETC --其他塑料垃圾Styrofoam_Buoy --泡沫浮标垃圾 图片共2377张原始图片。数据集按8:1:1分配。其中 训练集1901张图片验证集238张图片测试集238张。图片中包含噪声和亮度增强图片像素2048*20481使用YOLOv8对水面漂浮垃圾数据集进行目标检测11种类别的垃圾目标并有2377张原始图片按照8:1:1的比例分配为训练集、验证集和测试集。一、环境准备安装CUDA驱动GPU的NVIDIA CUDA驱动程序。通过运行nvidia-smi命令来检查是否已正确安装。安装Anaconda并创建虚拟环境首先下载并安装Anaconda。然后创建一个新的Python虚拟环境conda create-nmarine_debris_detectionpython3.8conda activate marine_debris_detection安装依赖项在激活的虚拟环境中安装必要的库pipinstalltorch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 pipinstallultralytics opencv-python matplotlib tqdm请根据你的CUDA版本调整PyTorch的安装链接。二、数据集组织数据集按如下结构组织YOLOv8支持YAML格式的数据配置marine_debris_dataset/ ├── images/ │ ├── train/ │ ├── val/ │ └── test/ ├── labels/ │ ├── train/ │ ├── val/ │ └── test/ └── data.yaml其中data.yaml文件应包含如下信息train:./marine_debris_dataset/images/trainval:./marine_debris_dataset/images/valtest:./marine_debris_dataset/images/testnc:11# 类别数量names:[Glass,Styrofoam_Piece,Plastic_Buoy_China,Metal,Net,Rope,Plastic_Buoy,PET_Bottle,Styrofoam_Box,Plastic_ETC,Styrofoam_Buoy]三、使用YOLOv8训练模型加载预训练模型fromultralyticsimportYOLO# 加载官方YOLOv8n预训练模型modelYOLO(yolov8n.pt)训练模型编写简单的Python脚本来启动训练过程deftrain_yolo():modelYOLO(yolov8n.yaml)# 使用YOLOv8n配置文件resultsmodel.train(data./marine_debris_dataset/data.yaml,epochs100,imgsz2048,# 根据你的图像大小调整batch8,# 考虑到2048*2048分辨率可能需要降低batch sizenamemarine_debris_train,projectruns/detect,saveTrue,save_period10,device0iftorch.cuda.is_available()elseNone,augmentTrue# 如果需要数据增强)if__name____main__:train_yolo()注意由于你的图像分辨率为2048x2048建议适当减少batch大小以适应显存限制。同时可以开启数据增强(augmentTrue)以提升模型泛化能力。四、推理代码用于对新的图像进行检测fromultralyticsimportYOLOimportcv2importosdefdetect_image(image_path,output_dirruns/detect/results):ifnotos.path.exists(output_dir):os.makedirs(output_dir)modelYOLO(runs/detect/marine_debris_train/weights/best.pt)# 替换为你的最佳权重路径resultsmodel.predict(sourceimage_path,saveTrue,projectoutput_dir,name)forrinresults:im_arrayr.plot()imcv2.cvtColor(im_array,cv2.COLOR_BGR2RGB)output_pathos.path.join(output_dir,os.path.basename(image_path))cv2.imwrite(output_path,im)print(f检测完成结果保存在{output_path})if__name____main__:detect_image(path/to/test/image.jpg)五、评估代码使用YOLOv8内置的评估函数直接对验证集进行评估fromultralyticsimportYOLOdefevaluate_yolo():modelYOLO(runs/detect/marine_debris_train/weights/best.pt)metricsmodel.val()# 自动使用验证集进行评估print(fmAP50:{metrics.box.map50:.3f})print(fmAP50-95:{metrics.box.map:.3f})if__name____main__:evaluate_yolo()
RELATED

相关推荐

GitHub周报:前端基本功回暖与AI落地实战拆解

GitHub周报:前端基本功回暖与AI落地实战拆解

这周GitHub上的趋势榜单扫下来,最明显的感觉是:前端在重新强调基本功,AI在加速往应用层落地。模型层的新教程还在涨,但增长速度已经被前端工具链、面试资源仓库,还有一批“帮业务把AI用起来”的框架项目超过。这篇周报…

📅 2026/11/8 16:17:49
Python单元测试与集成测试实战:从pytest到mock依赖隔离

Python单元测试与集成测试实战:从pytest到mock依赖隔离

开篇:测试是代码的“安全网”,不是KPI我写Python快十年,真正让我对“测试”这两个字产生敬畏的,不是你写了多少个测试用例、覆盖率数字多漂亮,而是某次深夜重构核心模块,跑完测试全绿的那一瞬间——那种“可…

📅 2026/11/8 16:17:49
车载AVB协议合规测试实战:基于Vector工具链的gPTP、SRP与AVTP验证指南

车载AVB协议合规测试实战:基于Vector工具链的gPTP、SRP与AVTP验证指南

车载网络工程师的一天,很多时候不是在测功能,而是在"证明"一个设备没毛病。AVB(Audio Video Bridging,音视频桥接)就是这样一个让人又爱又恨的协议族:它管着座舱里所有音视频流的同步和传输&…

📅 2026/10/18 9:21:20
MORE NEWS

更多资讯

📰

合肥电工证的办理内部消息:急需特种作业证才能上岗找工作

合肥电工证的办理内部消息:急需特种作业证才能上岗找工作 你是不是正盯着手机屏幕,心里打着鼓?是不是已经投了几份简历,却因为没有电工证被拒?现在市场上 急需特种作业证才能上岗找工作 ,特别是像合肥这样的城市, 电工证…

📰

上海哪里办高压电工证?真实经历告诉你怎么高效拿下证书

上海哪里办高压电工证?真实经历告诉你怎么高效拿下证书 你是不是正愁找不到工作,因为手里缺一个高压电工证?别急,我来帮你理清楚上海哪里办高压电工证的全流程。作为一名在驻马店做过项目现场管理员的老手,我经历过证书没办下来就上岗的尴尬,也帮不少人…

📰

榆林电工证培训价格值不值得考?怕考不过白交培训费

榆林电工证培训价格值不值得考?怕考不过白交培训费 怕考不过白交培训费?榆林电工证培训价格到底值不值得考?这是很多想报考特种作业操作证的学员最关心的问题。尤其是对于初次接触电工证考试的人来说,培训费动辄几千块,一不小心没考过,钱就打水漂了。今…

📰

白山市电工证号怎么换?实操考试心里没底怕挂科?电子证书怎么查?

白山市电工证号怎么换?实操考试心里没底怕挂科?电子证书怎么查? 实操考试心里没底怕挂科?白山市电工证号到期了怎么办?别急,这篇文章教你一套完整的流程和避坑指南,特别是针对刚拿到证或者准备考证的朋友, 电子证书…

📰

证书过期了不知道怎么复审延期?电工证考试标准费用及在哪考全攻略

证书过期了不知道怎么复审延期?电工证考试标准费用及在哪考全攻略 证书过期了不知道怎么复审延期?你不是一个人。很多人在特种作业证到期后才发现,复审流程复杂、费用不清、甚至不知道在哪考。今天我们就从电工证考试标准费用入手,给你讲明白怎么考、在哪…

📰

高级电工证谁给发工资?上岗必备的证书价值与薪资真相

高级电工证谁给发工资?上岗必备的证书价值与薪资真相 之前考的证在外省能不能转过来?这个问题困扰着不少在外地打工、准备转岗的电工朋友。如果你手头有高级电工证,却在外地工作,可能还在纠结这证能不能用、工资能不能拿到手。其实,…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

读完文章,想聊聊您的网站?

告诉我们您的行业与需求,资深顾问一对一梳理方案与报价,全程免费。

📞 💬