-
ResNet-18的一个简洁TensorFlow 2版本实现:resnet18-tf2
资源介绍
resnet18-tf2
TensorFlow ResNet的正式似乎未包括ResNet-18或ResNet-34。
该代码库提供了ResNet-18和ResNet-34的简单( )TensorFlow 2实现,直接从PyTorch的torchvision转换而来。 模型输出已经过验证,可与火炬视觉模型的输出以浮点精度匹配。
此代码已使用以下软件包版本进行了测试:
tensorflow==2.4.1
pytorch==1.2.0
torchvision==0.4.0