- 博客(4)
- 收藏
- 关注
原创 TypeError: ‘numpy.int64‘ object is not iterable ,‘int‘ object is not iterable
TypeError: 'numpy.int64' object is not iterable ,'int' object is not iterable
2022-12-07 21:52:16
1148
原创 注意力机制的实现-senet
注意力机制可以作为一个小技巧在我们的网络中实现,具体的实现方式如下(以pytorch)为例1.se-netsenet是典型的通道注意力机制,具体原理如下import torchimport torch.nn as nnclass se_block(nn.Module): def __init__(self, channel=512, ratio=16): super(se_block, self).__init__() self.avg_pool
2022-05-06 11:17:42
967
原创 Pytorch官方教程:快速上手训练自己的数据集
代码所使用的网络结构图:1.数据集下载#datesets.py(名称)import torchimport torchvisionimport torchvision.transforms as transformstransform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])batch_s..
2022-05-04 21:28:59
1161
原创 pytorch-grad-cam的实现
参考链接jacobgil/pytorch-grad-cam: Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Examples for classification, object detection, segmentation, embedding networks and more. Including Grad-CAM, Grad-CAM++, Score-CAM, A
2022-04-30 15:49:37
480
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人