视觉与自然语言处理:缺陷检测与假新闻识别
1. 视觉缺陷检测系统
1.1 上传TF模型到Vertex AI模型注册表
在模型训练完成并保存后,下一步是将模型工件上传到Vertex AI模型注册表。可以使用Vertex AI SDK或Google Cloud控制台UI来完成此操作。上传时,需要提供模型显示名称、上传区域、保存的模型工件的URI以及服务容器镜像。以下是相关配置和Python代码示例:
PROJECT_ID='417812395597'
REGION='us-central1'
ARTIFACT_URI='gs://my-training-artifacts/tf_model/'
MODEL_DISPLAY_NAME='tf-bangle-defect-detector-v1'
SERVING_IMAGE='us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest'
import aiplatform
aiplatform.init(project=PROJECT_ID, location=REGION)
model = aiplatform.Model.upload(
display_name=MODEL_DISPLAY_NAME,
artifact_uri=ARTIFACT_URI,
serving_container_image_uri=SERVING_IMAGE,
sync=True,
)
model.wait()
print("Model Display Name: ", model.display_n
超级会员免费看
订阅专栏 解锁全文
1895

被折叠的 条评论
为什么被折叠?



