【实验必用】使用ResNet18作为主干网络进行深度学习的Pytorch实现

使用ResNet18作为主干网络进行深度学习的Python实现

  • 一、ResNet18的Python复现
  • 二、获取ResNet18的预训练模型
  • 三、加载ResNet18预训练模型参数并设计自己的模型

一、ResNet18的Python复现

  下述代码(resnet18.py)源于NeurlPS2021的RUL论文,全名为“Relative Uncertainty Learning for Facial Expression Recognition”。

  如果你只是想用ResNet18网络作为自己的主干网络,那么直接将下述代码存入一个 resnet18.py 文件即可,完全不需要修改。

import torch
import torch.nn as nn


class BasicBlock(nn.Module

你可能感兴趣的:(代码复现,深度学习,pytorch,人工智能)