咖啡陪你

2022年5月8日

摘要：从CIFAR数据集制作开始教你训练自己的分类模型目录参考CIFAR的格式制作自己的数据集使用自己制作的数据集训练模型参考CIFAR的格式制作自己的数据集首先将所有图片按类别放在文件夹中，文件夹名为类别名。例如：存在20个类就分20个文件夹将所有图片的路径提取到一个文件中，文件中每行包含图阅读全文

posted @ 2022-05-08 19:42 咖啡陪你阅读(818) 评论(0) 推荐(0) 编辑

制作自己的python版本的类CIFAR10数据集

摘要：关于python 版本的CIFAR10的数据格式，官网上已经介绍： data – a 10000x3072 numpy array of uint8s. Each row of the array stores a 32x32 colour image. The first 1024 entries 阅读全文

posted @ 2022-05-08 19:29 咖啡陪你阅读(308) 评论(0) 推荐(0) 编辑

2022年4月27日

使用python代码画折线图【matplotlib】

摘要： 1、画折线图【一条示例】 import matplotlib.pyplot as plt import numpy as np x_axis_data = [1,2,3,4,5,6,7] #x y_axis_data = [68,69,79,71,80,70,66] #y plt.plot(x_ax 阅读全文

posted @ 2022-04-27 16:44 咖啡陪你阅读(5782) 评论(0) 推荐(1) 编辑

python画图折线图饼图柱状图

摘要：折线图 import numpy as np from matplotlib import pyplot as plt plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签 plt.reParams是一个配置表 plt.rcParams[ 阅读全文

posted @ 2022-04-27 16:42 咖啡陪你阅读(195) 评论(0) 推荐(0) 编辑

2022年4月26日

LaTeX多种颜色标注高亮修改内容

摘要： 1. 问题近日，需要在 LaTeX 文档中用多种颜色进行高亮标注。以往一般是生成 PDF 后手动标注，但实在是太低效，而且重新生成 PDF 以后还要重新标注。一番搜索后决定使用 soul 包来解决 2. 解决方法soul 的主要功能有四：字间空格 \so 下划线 \ul 删除线 \st 高亮 \h 阅读全文

posted @ 2022-04-26 16:54 咖啡陪你阅读(5042) 评论(0) 推荐(0) 编辑

2022年4月20日

AttributeError: 'numpy.float64' object has no attribute 'cuda'

摘要：这种类型的错误都是由于没有转换为tensor张量导致的，只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了，例如这样：resnet(Variable(torch.tensor(img).cuda())) 阅读全文

posted @ 2022-04-20 17:36 咖啡陪你阅读(898) 评论(0) 推荐(0) 编辑

2022年4月17日

摘要：文章目录概述一、利用torchstat 1.1 方法 1.2 代码 1.3 输出二、利用ptflops 2.1 方法 2.2 代码 2.3 输出三、利用thop 3.1 方法 3.2 代码 3.3 输出概述 Params：是指网络模型中需要训练的参数总数，理解为参数量。 FLOPs：是指浮点运算次数阅读全文

posted @ 2022-04-17 15:49 咖啡陪你阅读(2261) 评论(4) 推荐(0) 编辑

Pytorch测试模型的GFLOPs和Param大小

摘要： 1. 定义文件flops_counter.py import sys import torch import torch.nn as nn import numpy as np def get_model_complexity_info(model, input_res, print_per_lay 阅读全文

posted @ 2022-04-17 15:43 咖啡陪你阅读(862) 评论(0) 推荐(0) 编辑

计算pytorch模型FLOPs,Params

摘要： import torch from flops_counter import get_model_complexity_info input_shape = (3, 320, 568) split_line = '=' * 30 flops, params = get_model_complexit 阅读全文

posted @ 2022-04-17 15:42 咖啡陪你阅读(337) 评论(0) 推荐(0) 编辑

计算模型FLOPs和参数量

摘要：在pytorch环境下，有两个计算FLOPs和参数量的包thop和ptflops，结果基本是一致的。 thop 参考https://github.com/Lyken17/pytorch-OpCounter 安装方法：pip install thop 使用方法： from torchvision.mo 阅读全文

posted @ 2022-04-17 15:40 咖啡陪你阅读(2398) 评论(0) 推荐(0) 编辑

公告