可结合着一下两个看一下：
Pytorch：利用预训练好的VGG16网络提取图片特征
pytorch实现用VGG对图片的特征提取及可视化
对于自己训练好的网络，在参考以上两篇文章的基础上要根据自己的情况调整，
耐心去理解下原理就明白了。
详细解决方案
Pytorch---利用预训练好的神经网络提取图片特征
热度：31 发布时间：2023-12-14 12:24:12.0
相关解决方案
- MNIST 神经网络：准确率很低
- 了解 PyTorch 张量形状
- 神经网络. epoch， iteration， batchsize相关理解和说明
- pytorch 广播语义(Broadcasting semantics)
- 【Guide of Faster R-CNN】Build Faster RCNN in PyTorch
- pytorch MultiheadAttention 出现NaN
- 【pytorch】RuntimeError: Expected object of backend CPU but got backend CUDA for argument #2 ‘weight‘
- 【pytorch】BrokenPipeError: [Errno 32] Broken pipe
- Pytorch DDP一机多卡并行训练
- 深度学习PyTorch，TensorFlow中GPU利用率较低，CPU利用率很低，且模型训练速度很慢的问题总结与分析 Pytorch 并行训练（DP， DDP）的原理和应用
- NG 神经网络 构建网络基础
- PyTorch 模型训练实用教程（二）：数据
- pytorch、torch、torchvision介绍
- Pytorch---利用预训练好的神经网络提取图片特征
- Pytorch range()
- Pytorch optimizer.step()
- pytorch RuntimeError: input image (T: 9 H: 8 W: 2) smaller than kernel size (kT: 4 kH: 4 kW: 4)
- Pytorch-拼接与拆分
- Pytorch-索引与切片
- [备忘笔记]linux服务器+pytorch 常用命令行指令
- BERT实现关系分类抽取（pytorch）
- pytorch nn.Dropout使用
- Pytorch 手写数字识别
- pytorch model.load_state_dict报错
- Pytorch 清华源加速失效
- 手写数字识别 神经网络 C++ 实现（二：数据集准备工作）
- 手写数字识别 神经网络 C++ 实现（一：openCV 超基础矩阵操作）
- albert pytorch version
- pytorch 训练过程理解
- pytorch 模型保存与加载 cpu转GPU