![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
模型压缩
qq_35407318
模型压缩和加速,计算机视觉
展开
-
CUDA C编程
原创 2021-05-10 21:18:16 · 124 阅读 · 0 评论 -
int8量化总结
原创 2021-05-10 21:15:59 · 252 阅读 · 0 评论 -
pytorch得到模型的计算量和参数量
文章目录方法1方法2方法3方法4方法1pytorch自带方法,计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print("Number of parameter: %.2fM" % (total/1e6))方法2计算模型参数总量和模型计算量def count_params(model, input_size=224): # param_sum = 0 with open('原创 2020-10-29 14:38:14 · 17318 阅读 · 2 评论 -
pytorch出现的警告
遇到的警告:UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python numbertest_loss += F.nll_loss(output, target, size_average=False).data[0]解决方法:把data[0]改为item()test_loss +原创 2020-10-10 19:59:39 · 8733 阅读 · 6 评论 -
热图heatmap
热图定义:作为目前最常见的一种可视化手段,热图因其丰富的色彩变化和生动饱满的信息表达被广泛应用于各种大数据分析场景。原创 2020-10-08 16:15:05 · 250 阅读 · 0 评论 -
tensorflow2报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution......
笔记报错解决方法效果报错tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.解决方法在程序开头加入如下代码:physical_de转载 2020-08-05 19:24:39 · 4884 阅读 · 19 评论 -
pytorch读取自制数据集出现过的问题
笔记问题1问题描述:解决方式问题2问题描述:解决方式问题1问题描述:TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>解决方式数据格式不对, 把image转成tensor,参数transform进行如下设置就可以了:transform=transform.ToTensor()。注意检测原创 2020-07-13 23:10:47 · 1634 阅读 · 1 评论 -
pytorch读取自制数据集
读取自制数据集方法一方法2方法一数据集中的类别文件放置。import torchimport torchvisionimport torchvision.transforms as transforms# 读取数据集datasets = torchvision.datasets.ImageFolder(root="数据集路径", transform=transforms.Compose([ "数据预处理语句"比如 tra原创 2020-07-13 11:38:26 · 458 阅读 · 0 评论 -
Yolov3运行出现过的问题完美解决
笔记描述问题一解决方法问题2解决方法问题3解决方法描述对于一个初学者,想要跑通程序并不容易,因为经常报错不知道怎么解决。接下来是我跑yolov3遇到的问题,以及解决方法。我花了两天时间才解决,哎,这是初学者的痛。问题一问题:TypeError: ‘NoneType’ object is not subscriptable解决方法因为targets为None,说明读取labels路径错误,修改路径即可。问题2问题:UserWarning: indexing with dtype torch.原创 2020-07-08 08:54:10 · 2606 阅读 · 2 评论