#23: train_small.py refactoring using argparser

.gitignore

@@ -2,3 +2,4 @@
 *.tar
 -checkpoint.ipynb
 *-checkpoint.ipynb
+script/

train.py

@@ -16,7 +16,7 @@
 from utilities.utils import update_vis_plot
 from utilities.utils import visualize_GT
 
-from utilities.dataloader_test import detection_collate
+from utilities.dataloader import detection_collate
 import yolov1
 
 viz = visdom.Visdom()
@@ -54,16 +54,17 @@
 DATASET_PATH_MARTIN_LAP = "/home/martin/Desktop/5class/_class_balance/"
 DATASET_PATH_MARTIN_COM = "/home/martin/Desktop/_class_balance/"
 DATASET_PATH_JAEWON = "D:\dataset\VOC2012"
-# transforms.ToTensor()
-train_dataset = VOC(root = DATASET_PATH_MARTIN_COM,
-                    transform=None, cls_option = False, selective_cls=None)
+
+
+train_dataset = VOC(root = DATASET_PATH_MARTIN_COM, transform=None)
 
 train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
                                            batch_size = batch_size,
                                            shuffle = True,
                                            collate_fn=detection_collate)
 
 
+"""
 net = yolov1.YOLOv1()
 # visualize_weights_distribution(net)
 
@@ -155,4 +156,6 @@
                 'optimizer': optimizer.state_dict(),
             }, False, filename='checkpoint_{}.pth.tar'.format(epoch))
 
-    f.close()
+    f.close()
+
+"""