
Pytorch
文章平均质量分 84
PyTorch框架的使用
曾小蛙
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
原始U-Net模型代码
论文:1505.U-Net: Convolutional Networks for Biomedical Image Segmentation代码: https://github.com/milesial/Pytorch-UNet/blob/master/unet/unet_model.py每个蓝色框对应一个多通道特征图。通道的数量在框的顶部表示。x-y 大小在框的左下角提供。白盒代表复制的特征图。箭头表示不同的操作原创 2024-05-20 18:13:26 · 3056 阅读 · 0 评论 -
【图像分类】基于pytorch训练自己的分类模型、并使用flask部署
基于pytorch的简易分类模型,数据集直接可以用文件夹形式产生原创 2023-05-04 18:25:29 · 553 阅读 · 0 评论 -
NCCL error in: /pytorch/torch/lib/c10d/ProcessGroupNCCL ,unhandled cuda error, NCCLversion 2.7.8
结论pytorch 、cudatoolkit、cuda驱动的版本需一致问题描述使用多GPU训练 stylegan3 模型时:python train.py --outdir=training-runs --cfg=stylegan3-r \--data=datastes/your_data.zip \--cfg=stylegan3-r --gpus=4 --batch=32 --gamma=8 --kimg=1800 --snap=50 --tick=2 报错信息torch.原创 2022-04-24 16:53:44 · 21056 阅读 · 0 评论 -
【GANS】对Stylgan2的动漫头像预训练模型进行【属性编辑】
核心参考Pretrained Anime StyleGAN2 — convert to pytorch and editing images by encoder下载项目相关stylegan2-pytorch (非官方)git clone https://github.com/viuts/stylegan2_pytorch.git动漫头像(anime-face)模型下载2020-01-11-skylion-stylegan2-animeportraits-networksnapshot-024原创 2021-09-30 14:59:25 · 2743 阅读 · 9 评论 -
【图像搜索】基于pytorch官方预训练模型的简易相似图片搜索
简易的相似图像搜索算法图片数据库查询结果本文主要方法流程预训练模型 + 单张图像的特征 ( 逐个保存,形成数据库)+ 特征压缩(选做)获取查询图像的特征向量将查询的特征向量与数据库保存的所有特征进行余弦距离计算返回结果实际编写自定义图片数据集读取代码pytorch SWAV预训练模型 (paper Unsupervised Learning of Visual Features by Contrasting Cluster Assignments.)抽原创 2021-05-31 19:59:15 · 7189 阅读 · 2 评论 -
转置卷积的理解:基于Paddle、Pytorch、Tensorflow相关API的浅显理解
转置卷积(Transposed Convolution)、反卷积(Deconvolution)、微步卷积(Fractional strided convolution)、后向卷积(backwards convolution)、亚像素卷积(sub-pixs convolution)一般而言 转置卷积≈反卷积≈微步卷积≈后向卷积≈亚像素卷积【from】Is the deconvolution layer the same as a convolutional layer?A note on RealTi原创 2020-12-22 22:02:03 · 661 阅读 · 0 评论