change import name

This commit is contained in:
Lorenzo 2021-03-22 10:00:46 +01:00
parent 292ea8a21a
commit 34fa7a5933
3 changed files with 6 additions and 6 deletions

View File

@ -9,10 +9,10 @@ import torch
from PIL import Image
from sklearn.metrics import accuracy_score
from monstereo.network import Loco
from monstereo.network.process import factory_for_gt, preprocess_pifpaf
from monstereo.activity import social_interactions
from monstereo.utils import open_annotations, get_iou_matches, get_difficulty
from monoloco.network import Loco
from monoloco.network.process import factory_for_gt, preprocess_pifpaf
from monoloco.activity import social_interactions
from monoloco.utils import open_annotations, get_iou_matches, get_difficulty
class ActivityEvaluator:

View File

@ -283,7 +283,7 @@ class PreprocessKitti:
def prep_activity(self):
"""Augment ground-truth with flag activity"""
from monstereo.activity import social_interactions
from monoloco.activity import social_interactions
main_dir = os.path.join('data', 'kitti')
dir_gt = os.path.join(main_dir, 'gt')
dir_out = os.path.join(main_dir, 'gt_activity')

View File

@ -1,7 +1,7 @@
from setuptools import setup
# extract version from __init__.py
with open('monstereo/__init__.py', 'r') as f:
with open('monoloco/__init__.py', 'r') as f:
VERSION_LINE = [l for l in f if l.startswith('__version__')][0]
VERSION = VERSION_LINE.split('=')[1].strip()[1:-1]