- 博客(5)
- 收藏
- 关注
原创 获取单个进程所占用的CPU、GPU以及时间,python库psutil的用法
我们在模型训练过程中,有时候会把CPU、GPU给跑满掉,这将会使我们的模型中断,非常痛苦,因此我们需要一些工具来监测当前进程的CPU、GPU占用情况,并使用log记录下来。
2024-07-01 15:30:37
199
原创 【COCO数据集与VOC数据集格式互相转换】
很多时候我们跑代码时会发现,若不将数据集的格式进行转化,很难在特定的代码上获取所需的数据。于是,将不同的数据集进行格式转换是十分必要的。以下是关键部分的代码,需要完整代码的可以查看评论区留言。COCO数据集与VOC数据集的格式有所不同,包括训练集和验证集的存放方式以及标签的格式。VOC的标注格式采用的是.xml,训练集与验证集全部放在JPRGImages文件夹下。COCO的标注格式采用的是.txt格式,而图片的格式是以.jpg存放。
2024-06-29 14:21:41
219
使用YOLOV8训练自己的数据集,获取更详细的map指标,map-small,map-medium,map-large等
2024-06-29
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人