김성주

updated dataset

import os
import tensorflow as tf
tf.compat.v1.enable_eager_execution()
all_path = '../data/annotation.txt'
train_path = '../data/train.txt'
val_path = '../data/val.txt'
......
......@@ -15,7 +15,7 @@ def readImage(path):
return byte
def main():
ANNOTATION_PATH = 'train.txt' #annotation set (train/val/test) text file
ANNOTATION_PATH = '../data/train.txt' #annotation set (train/val/test) text file
IMAGE_DIRECTORY = 'image_data/' #image directory
SAVE_PATH = 'train.tfrecord' #save path for tfrecord
......
......@@ -11,4 +11,4 @@ so let me know if there are some errors/problems in python code version
3] fixed saver to save the parameter only when save-optimizer option is true
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')
5] wrote TFRecord binary iterator, which runs without tf session (references: https://github.com/pgmmpk/tfrecord )
6] removed logging/tenorboard summary code. (I will add it later if necessary)
\ No newline at end of file
6] removed logging/tensorboard summary code. (I will add it later if necessary)
\ No newline at end of file
......
......@@ -13,7 +13,7 @@ from data_utils import letterbox_resize
from model import yolov3
parser = argparse.ArgumentParser(description="YOLO-V3 test single image test procedure.")
parser.add_argument("input_image", type=str,
parser.add_argument("--input_image", type=str,
help="The path of the input image.")
parser.add_argument("--anchor_path", type=str, default="../../data/yolo_anchors.txt",
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.