使用ResNet18作为主干网络进行深度学习的Python实现
一、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