计算机视觉
文章平均质量分 50
全幼儿园最聪明
计算机专业硕士在读
展开
-
python 写 csv 文件的两种方法
文章目录问题描述方法一: csv方法二: pandas问题描述在深度学习相关任务的训练时,需要在训练的每个 epoch 记录当前 epoch 的准确率(如下图所示),那么在 python 中要怎么将内容写入 csv 文件呢,学习发现可以使用 csv 或者 pandas 实现,在这里做个简单记录。这里示例的代码为以追加模式写,每次写入一行方法一: csvimport csvlog_path = 'log/temp.csv'file = open(log_path, 'a+', encoding原创 2022-04-23 20:40:09 · 18393 阅读 · 0 评论 -
解决 ImportError: cannot import name ‘Flatten‘ from ‘torch.nn‘
问题描述在代码中从torch.nn引入Flatten报错:解决方案进入python交互式环境输入以下代码,查看pytorch版本:import torchprint(torch.__version__)可以看到pytorch版本是1.1.0版本太低不包含Flatten属性因此,一个办法就是升级pytorch版本,但我们知道下载安装pytorch并不容易,若没有镜像下载缓慢且需要考虑显卡驱动、conda、torch三者的版本关系,升级可能导致版本不匹配conda无法使用gpu等问题。原创 2022-02-11 00:11:06 · 5231 阅读 · 0 评论 -
如何理解深度学习中迁移学习、预训练、微调的概念
说明本文为观看吴恩达深度学习课程视频所做的学习笔记吴恩达 deeplearning.aiPart 3 结构化机器学习项目第二周 机器学习(ML)策略(2)2.7迁移学习正文假设已经训练好一个图像识别神经网络,如果让这个图像识别神经网络适应或者说迁移不同任务中学到的知识,比如放射科诊断(阅读X射线扫描图)。做法是:删掉神经网络最后的输出层和进入到最后一层的权重为最后一层重新赋予随机权重让神经网络在放射诊断数据上训练这样把图像学习的知识应用或迁移到放射科诊断上来,这就是迁移学习(Tran原创 2021-12-09 23:50:58 · 2140 阅读 · 1 评论