【发布时间】:2021-03-31 19:18:06
【问题描述】:
要在 ModelZoo 中使用 TensorFlow 官方模型中的任何对象检测模型,有一个名为“VAL_JSON_FILE”的变量,用于params_override 参数。对于我的用例,我在 RetinaNet 上执行迁移学习。命令和参数如下:
TPU_NAME="<your GCP TPU name>"
MODEL_DIR="<path to the directory to store model files>"
RESNET_CHECKPOINT="<path to the pre-trained Resnet-50 checkpoint>"
TRAIN_FILE_PATTERN="<path to the TFRecord training data>"
EVAL_FILE_PATTERN="<path to the TFRecord validation data>"
VAL_JSON_FILE="<path to the validation annotation JSON file>"
python3 ~/models/official/vision/detection/main.py \
--strategy_type=tpu \
--tpu="${TPU_NAME?}" \
--model_dir="${MODEL_DIR?}" \
--mode=train \
--params_override="{ type: retinanet, train: { checkpoint: { path: ${RESNET_CHECKPOINT?}, prefix: resnet50/ }, train_file_pattern: ${TRAIN_FILE_PATTERN?} }, eval: { val_json_file: ${VAL_JSON_FILE?}, eval_file_pattern: ${EVAL_FILE_PATTERN?} } }"
这是链接:https://github.com/tensorflow/models/blob/master/official/vision/detection/README.md
VAL_JSON_FILE 是做什么用的?有必要吗?如何找到/创建文件?
谢谢!
【问题讨论】:
标签: python tensorflow object-detection-api