AIPlatform_6787/TrainSdk/Test.py

import TrainSdk
import json

#获取训练文件夹中的数据数量, 测试token必须使用 4925EC4929684AA0ABB0173B03CFC8FF。
file_count = TrainSdk.get_file_count("4925EC4929684AA0ABB0173B03CFC8FF")
#打印数量
print("Current file count: " + str(file_count))
#循环读取文件和标注结果
for index in range(file_count):
    print("Reading file: " + str(index) + "...")
    file_data, label_data, file_name, file_isVideo = TrainSdk.get_labeled_file("4925EC4929684AA0ABB0173B03CFC8FF",index)
    #输出获取的文件
    print(file_data)
    #输出获取的标注结果
    labeldata = json.loads(label_data)
    print(labeldata)
    #输出获取的文件名称
    print(file_name)
    #输出获取的文件是否是视频
    print(file_isVideo)
    #保存模型
    TrainSdk.save_output_model("d:\test.pb")