From bcecaa52ae42c9379ce31d21638c1a02d11e97be Mon Sep 17 00:00:00 2001 From: nima <54316106+nimadorostkar@users.noreply.github.com> Date: Sun, 17 Sep 2023 10:47:55 +0330 Subject: [PATCH] Update load_weights.py --- load_weights.py | 3 +++ 1 file changed, 3 insertions(+) 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)