- 博客(7)
- 收藏
- 关注
RSS订阅原创 Python脚本统计列表元素
numbers = [1,2,4,2,3,5,2,4,2,3]# 例如 set01 = set(numbers) # 将列表转化为集合，因为列表元素可重复，集合不可以，所以首先统计出来多少种元素 print(set01)# 输出集合 dict01 = {}# 初始化一个空字典 for item in set01: dict01.up...
2020-04-29 20:16:07
217
原创 RuntimeError: Error(s) in loading state_dict for ResNet:
对于这个问题，我的问题需要修改的是修改前：net.load_state_dict(torch.load(path_net))修改后：net.load_state_dict(torch.load(path_net),False)具体原因见函数声明： def load_state_dict(self, state_dict, strict=True): ...
2020-04-10 23:05:27
6885
原创 德国交通标志识别训练集、验证集的划分，测试集读取
需求：读取csv文件的最后一列，真是底子不实在，曲线救国啊！先读出来的是字符串的列表，不知道为何把表头也读进来了，然后把他踢出去（出栈），然后再转化为整型的列表。学会把每一步输出，学会debug！import csvdef read_label(dir): with open(dir, "r") as f: #"D:\\dataset\\GTS...
2020-04-09 15:47:15
2078
原创 RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 54 and 45 i
查阅了各种博客，资料，其实就是细节问题：表现在batchsize、图片尺寸的问题。我是在测试集的加载出了问题，没将测试集的图片尺寸设置统一。
2020-04-09 15:42:03
262
原创 输出model的具体层及细节设置--torch实现
def resnet32(): 返回一个resnet20的类if __name__ == "__main__": for net_name in __all__: if net_name.startswith('resnet'): #print(net_name) test(globals()[net_name...
2020-04-09 08:47:17
868
原创 全局平均池化---torch实现
# target output size of 5x7m = nn.AdaptiveAvgPool2d((5,7))input = torch.randn(1, 64, 8, 9)output = m(input)# output.shape=(1,64,5,7)# target output size of 7x7 (square)m = nn.AdaptiveAvgPool2...
2020-04-06 23:48:30
4167
原创 数据集的图片尺寸与想用的网络默认输入尺寸不匹配怎么办？--torch实现
修改数据集图片尺寸以适应现有网络的输入？这是个方法：model_vgg = vgg(include_top=False, weights='imagenet', input_shape=(48, 48, 3))# 选择imagnet,会选择当年大赛的初始参数# include_top=False 去掉最后3层的全连接层看源码可知...
2020-04-04 16:05:46
7717
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人