Update load_weights.py

nimadorostkar · Sep 17, 2023 · bcecaa5 · bcecaa5
1 parent 7454c2b
commit bcecaa5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)