计算机视觉
tangjxycu
这个作者很懒,什么都没留下…
展开
-
pytorch修改网络结构后如何加装预训练模型
如果不知道如何使用预训练模型训练的朋友,使用resnet18模型训练自己的数据集的教程请看这篇博客一、如果我们网络没任何修改还是训练imagenet的数据集,1000个类别,那么只需以下代码:model_ft = models.resnet18(pretrained=True)model_ft = model_ft.to(device)二、如果只修改训练自己数据集类别如果我们训练自己的数据集,假如自己的数据集只有10个类别,那么需先加载预训练模型,然后修改最后的全连接层,改成以下代码加转载 2021-05-19 19:37:34 · 2157 阅读 · 3 评论 -
计算机视觉系列之一文看尽 GAN 的前世今生
当 Ian Goodfellow 在 2014 年喝了一杯啤酒之后,在梦中产生了「生成对抗网络」(GANs)的想法时,他可能没想到这一领域进展如此之快:转载 2020-11-30 19:49:14 · 1115 阅读 · 0 评论