From 18512ce5c3d97d573d51e39ddf2a63767de99160 Mon Sep 17 00:00:00 2001
From: "GiCheol.Kim" <37562319+gitcheol@users.noreply.github.com>
Date: Wed, 30 Jun 2021 17:17:03 +0900
Subject: [PATCH 1/2] fix the type error

there are a some small error when we execute this shell file.
---
 ML/Pytorch/object_detection/YOLO/data/get_data | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ML/Pytorch/object_detection/YOLO/data/get_data b/ML/Pytorch/object_detection/YOLO/data/get_data
index c7368b10..4970f6fb 100755
--- a/ML/Pytorch/object_detection/YOLO/data/get_data
+++ b/ML/Pytorch/object_detection/YOLO/data/get_data
@@ -7,11 +7,11 @@
                                                               
 ## OR DOWNLOAD FROM HERE (FASTER DOWNLOAD)                                          
 # VOC2007 DATASET                                                              
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.ta
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar # 
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
 
 # VOC2012 DATASET                                                              
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.ta
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
 
 # Extract tar files
 tar xf VOCtrainval_11-May-2012.tar

From 99cbc7e901e50cca4d664295f6a23ccc732c9102 Mon Sep 17 00:00:00 2001
From: "GiCheol.Kim" <37562319+gitcheol@users.noreply.github.com>
Date: Wed, 30 Jun 2021 18:04:54 +0900
Subject: [PATCH 2/2] Update train.py

---
 ML/Pytorch/object_detection/YOLO/train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ML/Pytorch/object_detection/YOLO/train.py b/ML/Pytorch/object_detection/YOLO/train.py
index 6908e8f4..19f27c42 100755
--- a/ML/Pytorch/object_detection/YOLO/train.py
+++ b/ML/Pytorch/object_detection/YOLO/train.py
@@ -36,8 +36,8 @@
 PIN_MEMORY = True
 LOAD_MODEL = False
 LOAD_MODEL_FILE = "overfit.pth.tar"
-IMG_DIR = "data/images"
-LABEL_DIR = "data/labels"
+IMG_DIR = "data/data/images"
+LABEL_DIR = "data/data/labels"
 
 
 class Compose(object):
@@ -84,7 +84,7 @@ def main():
         load_checkpoint(torch.load(LOAD_MODEL_FILE), model, optimizer)
 
     train_dataset = VOCDataset(
-        "data/100examples.csv",
+        "data/train.csv",
         transform=transform,
         img_dir=IMG_DIR,
         label_dir=LABEL_DIR,