【CANN训练营第三季】应用开发

应用开发流程

模型输入与输出
StatGAN的输入为128*128*3的图片和五个属性设置；输出为一张128*128*3的图片。
为了满足模型要求的输入，数据预处理中需要首先在图片中裁剪178*178大小的区域再Resize到128*128，之后作归一化操作。
输出中需要将数据范围由[-1,1]转换到[0,1]以及进行通道调整。
模型转换
需要将PyTorch模型转换为昇腾AI处理器支持的.om格式离线模型，使用的工具为ATC（Ascend Tensor Compiler）。

对于StarGAN而言，原始模型为onnx模型，转换之前需要进行转换分析，确定输入形状和格式，确定色域转换和归一化操作。

转换模型之后，可以利用msame工具进行测试以判断模型的性能及推理是否正常。