2024“大运河杯”数据开发应用创新大赛——城市治理

m0_58854572

已于 2024-09-02 16:06:55 修改

阅读量153

点赞数 1

文章标签：深度学习人工智能

于 2024-09-02 15:32:08 首次发布

本文链接：https://blog.csdn.net/m0_58854572/article/details/141820743

版权

比赛网址：

初赛还有10天+现在冲还来得及！

厚德云提供50元算力支持，够用了！

厚德云上的环境也配置好了，除了要修改一下数据下载地址其他的都很齐全，跑通baseline不成问题！

1、CUDA：out of memory，把模型从yolov8n换到yolov8l或x可能会出现这个问题，换成m暂时不会，清理缓存可以暂时解决，换一个更好的机器也可以解决，嘎嘎嘎

model = YOLO("yolov8m.pt")

import torch, gc
gc.collect()
torch.cuda.empty_cache()

2、data augmentation

results = model.train(data="yolo-dataset/yolo.yaml", epochs=2, imgsz=1088, batch=16, mixup=0.4, copy_paste=0.5, mosaic=0.9, erasing=0.6)

mixup以及后面的参数目前我是乱试，跑一个epoch要17mins左右，暂时不设置太大的epoch。

昨天从n换到m以后表现变好了，但是昨天没什么时间就只跑了一个epoch，今天同样的代码跑了6个epoch以为模型表现会变好（等了差不多2小时），结果和昨天一样，增大epoch并没有变好，于是今天试试数据增强。

## 破案了，我没有修改测试输入的地址，可能是把昨天的结果又上传了一遍，待会训练完这一轮，我把之前增大epoch的结果重新上传，刚好今天还剩两次提交机会

3、我注意到比赛还提供了没有标注的视频数据，不知道为什么排行榜上第一名的分数那么高是怎么做到的，这两天把datawhale提供的上分思路实践一下，之后我再找其他方法！

今天最后一次训练，还没有上传结果，先记下来！

关注