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 PIL import Image
from sklearn.metrics import accuracy_score from sklearn.metrics import accuracy_score
from monstereo.network import Loco from monoloco.network import Loco
from monstereo.network.process import factory_for_gt, preprocess_pifpaf from monoloco.network.process import factory_for_gt, preprocess_pifpaf
from monstereo.activity import social_interactions from monoloco.activity import social_interactions
from monstereo.utils import open_annotations, get_iou_matches, get_difficulty from monoloco.utils import open_annotations, get_iou_matches, get_difficulty
class ActivityEvaluator: class ActivityEvaluator:

View File

@ -283,7 +283,7 @@ class PreprocessKitti:
def prep_activity(self): def prep_activity(self):
"""Augment ground-truth with flag activity""" """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') main_dir = os.path.join('data', 'kitti')
dir_gt = os.path.join(main_dir, 'gt') dir_gt = os.path.join(main_dir, 'gt')
dir_out = os.path.join(main_dir, 'gt_activity') dir_out = os.path.join(main_dir, 'gt_activity')

View File

@ -1,7 +1,7 @@
from setuptools import setup from setuptools import setup
# extract version from __init__.py # 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_LINE = [l for l in f if l.startswith('__version__')][0]
VERSION = VERSION_LINE.split('=')[1].strip()[1:-1] VERSION = VERSION_LINE.split('=')[1].strip()[1:-1]