pipeline.config

pipeline.config

model {
  ssd {
    num_classes: 3
...
train_config {
  batch_size: 24
  fine_tune_checkpoint: "gs://edge-tpu-demo/model.ckpt"
  freeze_variables: [
    'FeatureExtractor/MobilenetV2/Conv',
    'FeatureExtractor/MobilenetV2/expanded_conv_1',
    'FeatureExtractor/MobilenetV2/expanded_conv_2',
    'FeatureExtractor/MobilenetV2/expanded_conv_3',
    'FeatureExtractor/MobilenetV2/expanded_conv_4',
    'FeatureExtractor/MobilenetV2/expanded_conv_5',
    'FeatureExtractor/MobilenetV2/expanded_conv_6',
    'FeatureExtractor/MobilenetV2/expanded_conv_7']
...
train_input_reader {
  label_map_path: "gs://edge-tpu-demo/label_map.pbtxt"
  tf_record_input_reader {
    input_path: "gs://edge-tpu-demo/train.record"
  }
}
...
eval_input_reader {
  label_map_path: "gs://edge-tpu-demo/label_map.pbtxt"
  shuffle: false
  num_readers: 1
  tf_record_input_reader {
    input_path: "gs://edge-tpu-demo/test.record"
  }
}
...