使用 ResNet-50 进行图像分类答案

【问题标题】：Image classification using ResNet-50使用 ResNet-50 进行图像分类
【发布时间】：2020-06-09 07:59:47
【问题描述】：

我尝试使用以下代码训练 ResNet 50 进行图像分类：https://github.com/mlperf/training/blob/master/image_classification/tensorflow/official/resnet/imagenet_main.py，但在处理过程中，我在第 322 行中出现错误 seed = int(argv[1]) IndexError: list index out of range。我该如何解决这个错误？我正在使用 TensorFlow 2.0。

【问题讨论】：

标签： image classification resnet

【解决方案1】：

您的错误消息IndexError: list index out of range 表明列表argv 的长度

sys.argv包含您在命令行上传递的参数值 em>列表。第一个元素argv[0] 始终是您调用的脚本的名称。在您的情况下，这意味着您没有传递任何参数。

因此，查看代码，它会尝试读取 - 作为第一个命令行选项 - 一个种子，它可以用来初始化各种随机数生成器 (RNG)。所以你想要做的是调用你的脚本，比如：

python imagenet_main.py r4nd0ms33d

r4nd0ms33d 是一个随机值。

有时使用固定值显式播种 RNG 可能很有用，例如当您希望您的训练结果可重现时。

你可以玩这个简单的 Python 程序：

import sys                                                                      

if __name__ == '__main__':                                                      
    print(sys.argv)

将其保存到987654328 @并查看它打印的内容如果您在python test.py与python test.py random Versus python test.py random seed

【讨论】：