深度学习------tensorflow2.0:ResNet50调库实现mnist、cifar10、cifar3数据集

1. resnet-50调库实现mnist数据集

读取mnist数据,直接输出切好的训练集和测试集,如果图片维度不同的话,需要转化维度,常见的转化维度方法有:squeeze(去除维度为1),reshape(转化为自己想要的维度),expand_dims(在任意位置加1,即升维)。
mnist数据集,图片维度为(28,28,1)

import tensorflow as tf
from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras import datasets
import matplotlib.pyplot as plt

猜你喜欢

转载自blog.csdn.net/m0_47405013/article/details/123295421