【华为云技术分享】从软件开发到 AI 领域工程师：模型训练篇 (5)

创建 Notebook 环境成功：

从 OBS 桶同步相关文件：

接下来进入该 Notebook 的终端环境，运行以下代码，启动训练任务：

训练过程输出片段如下所示：

2020-04-07 18:58:14.497319: I tensorflow/stream_executor/dso_loader.cc:152] successfully opened CUDA library libcublas.so.10.0 locally 7/7 [==============================] - 17s 2s/step - loss: 4226.4421 - val_loss: 22123.3750 Epoch 2/50 7/7 [==============================] - 6s 855ms/step - loss: 1083.1558 - val_loss: 1734.1427 Epoch 3/50 7/7 [==============================] - 6s 864ms/step - loss: 521.8567 - val_loss: 455.0971 Epoch 4/50 7/7 [==============================] - 6s 851ms/step - loss: 322.8907 - val_loss: 193.3107 Epoch 5/50 7/7 [==============================] - 6s 841ms/step - loss: 227.7257 - val_loss: 150.8902 Epoch 6/50 7/7 [==============================] - 6s 851ms/step - loss: 179.0605 - val_loss: 154.9351 Epoch 7/50 7/7 [==============================] - 6s 868ms/step - loss: 150.4297 - val_loss: 147.3101 Epoch 8/50 7/7 [==============================] - 8s 1s/step - loss: 129.5681 - val_loss: 144.8283

模型生成后，创建一个 python 脚本，代码如下，实现了模型文件拷贝到 OBS 桶：

from modelarts.session import Session session = Session() session.upload_data(bucket_path="/mask-detection-modelarts-test/run/log/", path="/home/ma-user/work/log/trained_weights_final.h5")

运行推理脚本，我把推测结果打印在了测试图片上，如下图所示，识别出了口罩：

后记

AI 技术的兴起，已经带动了科技行业的革命，而每一次业界的革命，都会让一些公司落寞而让另一些公司崛起，程序员也一样，每一次技术换代也都会让一些程序员没落而让另一些程序员崛起。抓住目前正在流行的 AI 技术趋势，使用云端的免费计算资源，上手学习并实践 AI 技术，会是相当一部分软件工程师、数据科学家的选择。此外，由于在图像识别、文本识别、语音识别等技术领域，算法的精度已经给有大幅度的提升，在很多场景下已经达到可商用级别，也进一步让自动机器学习技术（模型的自动设计和训练）的发展成为可能。因此，在上述几个技术领域的很多应用场景下，公有云已经可以做到根据用户自定义数据进行 AI 模型的自动训练。

原文地址：https://www.infoq.cn/article/0PKOlHjfs3CQWkqTAQ6T