Fix dependencies

embedding-calculator/src/services/facescan/plugins/adaface/data.py

@@ -4,10 +4,10 @@
 from torchvision import transforms
 import numpy as np
 import pandas as pd
-import evaluate_utils
-from dataset.image_folder_dataset import CustomImageFolderDataset
-from dataset.five_validation_dataset import FiveValidationDataset
-from dataset.record_dataset import AugmentRecordDataset
+import src.services.facescan.plugins.adaface.evaluate_utils
+from src.services.facescan.plugins.adaface.dataset.image_folder_dataset import CustomImageFolderDataset
+from src.services.facescan.plugins.adaface.dataset.five_validation_dataset import FiveValidationDataset
+from src.services.facescan.plugins.adaface.dataset.record_dataset import AugmentRecordDataset
 
 
 class DataModule(pl.LightningDataModule):

embedding-calculator/src/services/facescan/plugins/adaface/dataset/five_validation_dataset.py

@@ -1,6 +1,6 @@
 from torch.utils.data import Dataset
 import numpy as np
-import evaluate_utils
+import src.services.facescan.plugins.adaface.evaluate_utils
 import torch
 
 class FiveValidationDataset(Dataset):

embedding-calculator/src/services/facescan/plugins/adaface/inference.py

@@ -4,9 +4,6 @@
 import numpy as np
 from src.constants import ENV
 
-#with open('config.json', 'r') as f:
-#    config = json.load(f)
-
 device = ENV.DEVICE
 model_mame = ENV.RECOGNITION_MODEL
 model_path = ENV.RECOGNITION_MODEL_PATH

embedding-calculator/src/services/facescan/plugins/adaface/main.py

@@ -6,10 +6,10 @@
 from pytorch_lightning.callbacks import ModelCheckpoint
 from pytorch_lightning.loggers import CSVLogger, WandbLogger
 from pytorch_lightning import seed_everything
-import config
+import src.services.facescan.plugins.adaface.config
 import os
-from utils import dotdict
-import train_val
+from src.services.facescan.plugins.adaface.utils import dotdict
+import src.services.facescan.plugins.adaface.train_val as train_val
 import data
 import inspect
 

embedding-calculator/src/services/facescan/plugins/adaface/train_val.py

@@ -3,9 +3,9 @@
 import torch.optim.lr_scheduler as lr_scheduler
 from pytorch_lightning.core import LightningModule
 from torch.nn import CrossEntropyLoss
-import evaluate_utils
-import head
-import net
+import src.services.facescan.plugins.adaface.evaluate_utils as evaluate_utils
+import src.services.facescan.plugins.adaface.head as head
+import src.services.facescan.plugins.adaface.net as net
 import numpy as np
 import utils
 

embedding-calculator/src/services/facescan/plugins/adaface/validation_lq/validate_IJB_S.py

@@ -1,13 +1,13 @@
 import torch
 import numpy as np
 from tqdm import tqdm
-import data_utils
+import src.services.facescan.plugins.adaface.validation_lq.data_utils as data_utils
 import argparse
 import pandas as pd
 import evaluate_helper
 import sys, os
 sys.path.insert(0, os.path.dirname(os.getcwd()))
-import net
+import src.services.facescan.plugins.adaface.net as net
 
 
 def str2bool(v):