
深度学习
猫四问
新手上路
展开
-
【无标题】
ImageNet中文标签转载 2022-06-16 20:27:34 · 1074 阅读 · 0 评论 -
pytorch训练自己的数据集
使用pytorch自带的模型,并修改全连接层为自己数据集的类别数。model = models.resnet50(pretrained=False)class_num = 62fc_features = model.fc.in_featuresmodel.fc = nn.Linear(fc_features, class_num)model = model.to(device)根据数据集文件夹制作包含文件路径 标签的txt文本文件import osdata = []labels = []原创 2021-07-01 16:26:04 · 1117 阅读 · 0 评论 -
tensorflow使用踩坑
问题:AttributeError: ‘_TfDeviceCaptureOp’ object has no attribute ‘_set_device_from_string’解决办法:已经安装的tensorflow版本为gpu 1.14.0,卸载后重新安装了1.12.0,但是又出现了Failed to load tensorflow runtime,考虑是服务器cuda版本为10.0,所以重新安装tensorflow-gpu==1.13.1后暂时解决。问题:Keras使用多卡训练from ke原创 2021-05-18 16:43:00 · 764 阅读 · 0 评论 -
python中两幅图像相减造成的问题
文章链接https://blog.youkuaiyun.com/weixin_42954710/article/details/105813459cv2读取的图片相减后会出现很多0和255的情况使用 img.astype(np.int32) 后解决原创 2021-01-26 13:59:30 · 962 阅读 · 0 评论 -
关于图像resize的观察
CV2的resizehttps://blog.youkuaiyun.com/wzhrsh/article/details/101630396 默认方法 双线性插值 INTER_LINEARPIL.Image的resize默认方法为BICUBIC (int 3)pytorch的transform.Resize()输入一个参数时,将较小的缩放到该参数值,较大的按比例缩放两个参数时缩放到指定大小调用 PIL.Image.resize() ,但是默认方法为BILINEAR (int 2)...原创 2021-01-22 11:37:54 · 211 阅读 · 0 评论 -
PIL文档与CV2文档
cv2http://www.woshicver.com/https://www.kancloud.cn/aollo/aolloopencv/269602PILhttps://pillow.readthedocs.io/en/stable/原创 2021-01-22 10:51:29 · 289 阅读 · 0 评论 -
cv2的一些使用记录
读取图片读取为BGR通道cv2.imread()读取为RGB通道cv2.imread()[:, :, ::-1]分离通道cv2.split()显示图片cv2.imshow("窗口名称",img)imshow会把BGR图片按RGB显示同理会把RGB按BGR显示未完待续原创 2021-01-21 20:09:11 · 179 阅读 · 0 评论 -
Pytorch加载预训练模型时间过长
Pytorch加载预训练模型时间过长解决办法:Scipy包的版本为1.4.1,升级为最新的1.6.0后解决。原创 2021-01-21 18:44:10 · 1081 阅读 · 0 评论