首页 > 解决方案 > 如何理解用于 Imagenet 预处理的 TensorFlow 源代码

问题描述

我想了解和复制 tensorflow.keras 如何在函数tf.keras.applications.resnet.preprocess_input中对 imagenet 进行预处理,但我无法理解源代码

@keras_export('keras.applications.imagenet_utils.preprocess_input')
@keras_modules_injection
def preprocess_input(*args, **kwargs):
  return imagenet_utils.preprocess_input(*args, **kwargs)

该定义似乎是递归的。此外,文档没有说明该功能。我怎样才能看到这个功能真正在做什么?

标签: pythontensorflowkerasresnetimagenet

解决方案


我认为这个链接会有所帮助。https://github.com/keras-team/keras-applications/blob/master/keras_applications/imagenet_utils.py。该定义实际上不是递归的,因为 import 语句from keras_applications import imagenet_utils将允许您使用位于 keras 中的 imagenet_utils 中的 preprocess_input 函数,而不是调用自身。


推荐阅读