随笔分类 - AI
摘要：替换BN层为IN层 最近在做实验时，考虑将官方torchvision包中的Resnet模型进行一些更改，ResNet类中有个可选参数_norm_layer可以直接传入nn.InstanceNorm2d，默认为nn.BatchNorm，但是这样更改后，在使用官方的预训练权重时，会发生一些报错，BN层里
阅读全文
摘要：统计数据集中的mean和std（适用txt中包含所有图片路径） 如果图片全在一个目录下可以参考 统计数据集中的mean和std（适用数据集所有图片在同一个文件夹下） 代码： import os from PIL import Image import numpy as np import tqdm
阅读全文
摘要：np.unique函数获取mask图片中不同像素数 @Data 2022/12/9 代码： import os import numpy as np from PIL import Image def unique_values(root): imgs = os.listdir(root) conc
阅读全文
摘要：transforms.Pad边界填充 @Data 2022/12/9 代码： import os import torch import torchvision.transforms as transform from PIL import Image import matplotlib.pyplo
阅读全文
摘要：统计数据集中的mean和std（适用数据集所有图片在同一个文件夹下） 统计数据集中的mean和std（适用数据集所有图片在同一个文件夹下） 代码： import os from PIL import Image import numpy as np import tqdm def main(): #
阅读全文
摘要：numpy中array和asarray array和asarray都可以将结构数据转化为ndarray，但是主要区别就是当数据源是ndarray时，array会copy出一个副本，占用新的内存，属于深拷贝。asarray是浅拷贝。 import numpy as np a = np.array([1
阅读全文
摘要：Pytorch（GPU版本）+cuda10.2 文章目录 Pytorch（GPU版本）+cuda10.2版本及说明步骤0、Anaconda创建环境（创建了可跳过）1、访问网址找到对应需要版本的torch2、下载3、安装4、测试是否成功5、返回False的可能原因6、参考链接 版本及说明 使用Anac
阅读全文
