diff --git a/load_weights.py b/load_weights.py index 21462f6..ea8e443 100644 --- a/load_weights.py +++ b/load_weights.py @@ -4,11 +4,13 @@ from yolov3_tf2.models import YoloV3, YoloV3Tiny from yolov3_tf2.utils import load_darknet_weights + flags.DEFINE_string('weights', 'weights/yolov3.weights', 'path to weights file') flags.DEFINE_string('output', 'weights/yolov3.tf', 'path to output') flags.DEFINE_boolean('tiny', False, 'yolov3 or yolov3-tiny') flags.DEFINE_integer('num_classes', 80, 'number of classes in the model') +#ttt def main(_argv): if FLAGS.tiny: @@ -29,6 +31,7 @@ def main(_argv): logging.info('weights saved') + if __name__ == '__main__': try: app.run(main)