diff --git a/mmp/a3/annotation.py b/mmp/a3/annotation.py index c2fb123..4977868 100644 --- a/mmp/a3/annotation.py +++ b/mmp/a3/annotation.py @@ -1,4 +1,3 @@ -import glob import os import re from typing import List diff --git a/mmp/a3/dataset.py b/mmp/a3/dataset.py index 4db8e0c..cd38a1c 100644 --- a/mmp/a3/dataset.py +++ b/mmp/a3/dataset.py @@ -4,7 +4,7 @@ from PIL import Image from typing import Tuple import torch from torch.utils.data import DataLoader -from a3.annotation import read_groundtruth_file +from .annotation import read_groundtruth_file from torchvision import transforms diff --git a/mmp/a3/main.py b/mmp/a3/main.py index c87fa3b..bcdaf9c 100644 --- a/mmp/a3/main.py +++ b/mmp/a3/main.py @@ -1,7 +1,7 @@ import torch import argparse -from a2.main import MmpNet, get_criterion_optimizer, train_epoch, eval_epoch -from a3.dataset import get_dataloader +from ..a2.main import MmpNet, get_criterion_optimizer, train_epoch, eval_epoch +from .dataset import get_dataloader def main():