김성주

updated dataset

1 import os 1 import os
2 import tensorflow as tf 2 import tensorflow as tf
3 3
4 +tf.compat.v1.enable_eager_execution()
5 +
4 all_path = '../data/annotation.txt' 6 all_path = '../data/annotation.txt'
5 train_path = '../data/train.txt' 7 train_path = '../data/train.txt'
6 val_path = '../data/val.txt' 8 val_path = '../data/val.txt'
......
...@@ -15,7 +15,7 @@ def readImage(path): ...@@ -15,7 +15,7 @@ def readImage(path):
15 return byte 15 return byte
16 16
17 def main(): 17 def main():
18 - ANNOTATION_PATH = 'train.txt' #annotation set (train/val/test) text file 18 + ANNOTATION_PATH = '../data/train.txt' #annotation set (train/val/test) text file
19 IMAGE_DIRECTORY = 'image_data/' #image directory 19 IMAGE_DIRECTORY = 'image_data/' #image directory
20 SAVE_PATH = 'train.tfrecord' #save path for tfrecord 20 SAVE_PATH = 'train.tfrecord' #save path for tfrecord
21 21
......
...@@ -11,4 +11,4 @@ so let me know if there are some errors/problems in python code version ...@@ -11,4 +11,4 @@ so let me know if there are some errors/problems in python code version
11 3] fixed saver to save the parameter only when save-optimizer option is true 11 3] fixed saver to save the parameter only when save-optimizer option is true
12 4] changed parameter 'mode' to bool value 'is_training' in data util functions (string value 'mode' is passed as byte string, so functions do not evaluate if-clauses as expected. ex) 'train' != b'train') 12 4] changed parameter 'mode' to bool value 'is_training' in data util functions (string value 'mode' is passed as byte string, so functions do not evaluate if-clauses as expected. ex) 'train' != b'train')
13 5] wrote TFRecord binary iterator, which runs without tf session (references: https://github.com/pgmmpk/tfrecord ) 13 5] wrote TFRecord binary iterator, which runs without tf session (references: https://github.com/pgmmpk/tfrecord )
14 -6] removed logging/tenorboard summary code. (I will add it later if necessary)
...\ No newline at end of file ...\ No newline at end of file
14 +6] removed logging/tensorboard summary code. (I will add it later if necessary)
...\ No newline at end of file ...\ No newline at end of file
......
...@@ -13,7 +13,7 @@ from data_utils import letterbox_resize ...@@ -13,7 +13,7 @@ from data_utils import letterbox_resize
13 from model import yolov3 13 from model import yolov3
14 14
15 parser = argparse.ArgumentParser(description="YOLO-V3 test single image test procedure.") 15 parser = argparse.ArgumentParser(description="YOLO-V3 test single image test procedure.")
16 -parser.add_argument("input_image", type=str, 16 +parser.add_argument("--input_image", type=str,
17 help="The path of the input image.") 17 help="The path of the input image.")
18 parser.add_argument("--anchor_path", type=str, default="../../data/yolo_anchors.txt", 18 parser.add_argument("--anchor_path", type=str, default="../../data/yolo_anchors.txt",
19 help="The path of the anchor txt file.") 19 help="The path of the anchor txt file.")
......
This diff is collapsed. Click to expand it.
This file is too large to display.
This diff is collapsed. Click to expand it.
This file is too large to display.
This diff is collapsed. Click to expand it.
This file is too large to display.
This diff is collapsed. Click to expand it.