Added script for reformatting TF OD API config files (#3)

Merging Marcel's script.
This commit is contained in:
Marcel Simon 2019-04-16 04:34:59 +09:00 коммит произвёл Siyu Yang
Родитель 9e6d242e55
Коммит 3dd26acd13
1 изменённых файлов: 18 добавлений и 0 удалений

Просмотреть файл

@ -0,0 +1,18 @@
# Simple script for reformatting a pipeline configuration in a
# canonical format such that it is possible to use diff to compare
# two pipelines
# You can directly compare two files using:
# diff <(python reformat.py pipeline1.config) <(python reformat.py pipeline2.config)
# Remember to put the models/research directory in your PYTHONPATH before calling
# the script
from object_detection.protos import pipeline_pb2
import google.protobuf.text_format as txtf
import sys
pipeline = pipeline_pb2.TrainEvalPipelineConfig()
with open(sys.argv[1]) as f:
txtf.Merge(f.read(), pipeline)
print(pipeline)