<reponame>Trevol/keras-retinanet
import os
import sys
from keras_retinanet.bin.convert_model import main as convertModel

os.environ["CUDA_VISIBLE_DEVICES"] = ""


def main():
    args = [
        './snapshots/resnet50_csv_07.h5',  # model_in
        'resnet50_random_shapes_07_ellipses.h5'  # model_out
    ]
    convertModel(args)


if __name__ == '__main__':
    main()

import os
import sys
from keras_retinanet.bin.convert_model import main as convertModel

os.environ["CUDA_VISIBLE_DEVICES"] = ""


def main():
    args = [
        './snapshots/resnet50_csv_07.h5',  # model_in
        'resnet50_random_shapes_07_ellipses.h5'  # model_out
    ]
    convertModel(args)


if __name__ == '__main__':
    main()
# model_in # model_out