- 博客(11)
- 收藏
- 关注
原创 1,如何修改模型的某个层的参数?
1,获取模型参数的键值对?import torchmodel_path = r'./your_model'model_dicts = torch.load(model_path,map_location=torch.device('cpu'))print(model_dicts)for key,value in model_dicts.items(): print(key) """ modify values """model = create_your_mode
2022-04-07 22:14:35
609
原创 01.深度学习中GPU的多进程推理实现方式
1.需求场景一次需要推理的数据过多,需要使用多进程技术,同时推理,加快速度。2.技术实现2.1 进程:程序的基本执行实体,每一个进程都有它自己的地址空间,一般情况下,包括文本区域(text region)、数据区域(data region)和堆栈(stack region),是系统分配资源和调度的独立单位。动态性:进程的实质是程序在多道程序系统中的一次执行过程,进程是动态产生,动态消亡的。并发性:任何进程都可以同其他进程一起并发执行独立性:进程是一个能独立运行的基本单位,同时也是系统分配资源和
2022-01-12 17:13:51
4555
原创 01.csv模块写入问题
1.csv.writerow()函数在windows系统下存在写入空白行的问题import csvcsvfile = open('./test.csv','w')writer = csv.writer(csvfile)2.原因分析windows系統+writer都会在文件的一行的结尾加上一个进位负号,等于进行2次的enter符号3.问题修改import csvcsvfile = open('./test.csv','w',newline='')writer = csv.writer(cs
2021-12-29 14:09:35
538
原创 01.jupyter notebook 命令汇总
1 开始jupyter notebook --generate-configjupyter notebook --notebook-dir your_path # your path means path in strings
2021-12-22 18:52:39
512
原创 01.监控深度学习平台的运行情况
1.需求监控深度学习平台的运行情况2.方法watch -n 1 nvidia-smi3.影响因素Volatile GPU util调节batchsize和num_workers,使得该数值尽量达到100%Memory-Usage使得GPU内存尽量占满
2021-12-21 15:14:06
888
原创 04.计算ROC的auc面积的方法
1.背景计算一组列表的ROC代表的AUC的面积2.代码实现def get_auc(fpr, tpr): area = 0 fpr.sort() tpr.sort() fpr= [0] + fpr+ [1] tpr= [0] + tpr+ [1] for item in range(len(fpr)-1): area += (tpr[item]+tpr[item+1]) * 0.5 * (fpr[item+1]-fpr[item])
2021-12-20 17:01:42
162
原创 03.目标检测中计算两个列表的iou并返回逻辑判断值
1.需求计算预测框与标签框的iou大小,并判断是否满足某一阈值def get_iou(bbox1, bbox2, threshold=0.3): """ bbox1与bbox2为列表 """ a1 = (bbox1[2] - bbox1[0]) * (bbox1[3] - bbox1[1]) a2 = (bbox2[2] - bbox2[0]) * (bbox2[3] - bbox2[1]) xmin = max(bbox1[0], bbox2[0])
2021-12-17 17:03:47
638
原创 02.图像分割中将掩码转换为坐标点的方法(多目标)
场景图像分割中,模型输出二值掩码图,转化为目标轮廓的xy坐标点对集合实现该方法适用于单目标以及多目标在这里插入代码片
2021-12-17 11:25:00
3273
原创 01.图像分割中将掩码转换为坐标点的方法(单目标)
1.场景图像分割中,模型输出二值掩码图,转化为目标检测中的检测框坐标2.实现该方法只适用于单目标 import numpy as np def mask2box(self, mask): # [x1,y1,x2,y2] '''从mask反算出其边框 mask:[h,w] 0、1组成的图片 1对应对象,只需计算1对应的行列号(左上角行列号,右下角行列号,就可以算出其边框) ''' index = np.argwhe
2021-12-17 10:16:32
2116
原创 01.cv2解决读取含中文路径图片失败问题
cv2读取含有中文路径的图片失败import cv2import numpy as npdef read_cv(path): img_cv = cv2.imdecode(np.fromfile(path,dtype=np.uint8),1) return img_cv
2021-12-16 19:05:45
388
原创 01.numpy中判断None和nan方法
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题,有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能,丰富你的文章UML 图表FLowchart流程图导出与导入导出导入欢迎使用Markdown编辑器你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Mar
2021-12-16 12:21:51
3664
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人