导入NASNET并测试
引言
NASNET 是目前为止性能最好的图像分类网络,tensorflow/models 和yeephycho/nasnet-tensorflow已经公布了相应的训练和测试代码。但是提供的示例代码仅为一句命令行,封装程度太高,不便于理解和自己测试。为了实现利用尽量少的python代码自行测试,本人查阅源代码,并且在GitHub上搜寻相关问题,终于解决了这一问题。
实现
- 首先导入需要的库
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # 避免import tensorflow时的warning
import numpy as np
import matplotlib.pyplot