change : file structure modification

config/config.py

@@ -20,23 +20,5 @@ def load_args():
 
     return args
 
-def load_args_contrastive():
-    parser = argparse.ArgumentParser()
 
-    parser.add_argument('--epoches', type=int, default=3)
-    parser.add_argument('--save_inter', type=int, default=50)
-    parser.add_argument('--eval_inter', type=int, default=1)
-    parser.add_argument('--noise_threshold', type=int, default=0)
-    parser.add_argument('--lr', type=float, default=0.001)
-    parser.add_argument('--batch_size', type=int, default=64)
-    parser.add_argument('--model', type=str, default='ResNet18')
-    parser.add_argument('--model_type', type=str, default='duo')
-    parser.add_argument('--dataset_dir', type=str, default='../data/processed_data/npy_image/data_training')
-    parser.add_argument('--dataset_ref_dir', type=str, default='img_ref')
-    parser.add_argument('--output', type=str, default='../output/out_contrastive.csv')
-    parser.add_argument('--save_path', type=str, default='../output/best_model_constrastive.pt')
-    parser.add_argument('--pretrain_path', type=str, default=None)
-    args = parser.parse_args()
-
-    return args
 

image_ref/config.py

+import argparse
+
+
+def load_args_contrastive():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--epoches', type=int, default=3)
+    parser.add_argument('--save_inter', type=int, default=50)
+    parser.add_argument('--eval_inter', type=int, default=1)
+    parser.add_argument('--noise_threshold', type=int, default=0)
+    parser.add_argument('--lr', type=float, default=0.001)
+    parser.add_argument('--batch_size', type=int, default=64)
+    parser.add_argument('--model', type=str, default='ResNet18')
+    parser.add_argument('--model_type', type=str, default='duo')
+    parser.add_argument('--dataset_dir', type=str, default='../data/processed_data/npy_image/data_training')
+    parser.add_argument('--dataset_ref_dir', type=str, default='img_ref')
+    parser.add_argument('--output', type=str, default='../output/out_contrastive.csv')
+    parser.add_argument('--save_path', type=str, default='../output/best_model_constrastive.pt')
+    parser.add_argument('--pretrain_path', type=str, default=None)
+    args = parser.parse_args()
+
+    return args
\ No newline at end of file

image_ref/main.py

@@ -6,11 +6,11 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from config.config import load_args, load_args_contrastive
-from dataset.dataset_ref import load_data, load_data_duo
+from config import load_args_contrastive
+from dataset_ref import load_data, load_data_duo
 import torch
 import torch.nn as nn
-from image_ref.model import Classification_model_contrastive, Classification_model_duo_contrastive
+from model import Classification_model_contrastive, Classification_model_duo_contrastive
 import torch.optim as optim
 from sklearn.metrics import confusion_matrix
 import seaborn as sn