Arcofcosmos的博客

Python统计voc格式数据集类别数量分布并生成直方图

'''Author: TuZhouVersion: 1.0Date: 2021-09-13 10:04:38LastEditTime: 2021-09-13 10:36:17LastEditors: TuZhouDescription: FilePath: \python_test\traffic_dataset_clean.py'''import randomimport numpy as npdef clean_test(): with open('./test.txt'

使用yolo系列通常需要通过聚类算法生成anchors，本文给出kmeans以及kmeans++的python实现。数据格式为VOC的xml文件kmeans聚类'''Author: TuZhouVersion: 1.0Date: 2021-08-22 18:39:19LastEditTime: 2021-09-12 17:14:13LastEditors: TuZhouDescription: FilePath: \My_Mobile_Yolo2\utils\kmeans_for_anc

在做目标检测时，我个人比较喜欢VOC格式的数据集，所以遇到COCO格式的数据集习惯转为VOC的，再进行处理。先看coco格式转为voc格式需要修改的只有路径以及jsonFileName 这个列表，都已经标明了比较清晰的注释。'''Author: TuZhouVersion: 1.0Date: 2021-08-29 16:32:52LastEditTime: 2021-08-29 17:44:57LastEditors: TuZhouDescription: FilePath: \My_Y

使用pytorch训练模型时想要预先加载预训练模型，忽然出现这种错误。原因大概是该预训练模型保存方法是完全保存：torch.save(model, path)该方法将模型内容全部保存，甚至包括存放路径这导致将保存的模型换位置的后，load加载的时候可能导致路径出现问题解决方法：model = Model()scripted_module = torch.jit.script(model)torch.jit.save(scripted_module, 'pretrained_model.pt