Showing
11 changed files
with
5 additions
and
3 deletions
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.
-
Please register or login to post a comment