search

user0407 / CLUDA

Implementation of CLUDA: Contrastive learning in Unsupervised Domian Adaptation in Semantic Segmentation
23 stars 2 forks source link

ImportError: cannot import name 'centroid_cl' from 'mmseg.models.losses.contrastive_loss' #1

Closed yuheyuan closed 1 year ago

yuheyuan commented 2 years ago

Hi, When I r un run_train.sh Can not import name 'centroid_cl'

(yy_CLUDA) ailab@ailab:/media/ailab/data/yy/CLUDA$ ./run_train.sh 
Traceback (most recent call last):
  File "run_experiments.py", line 15, in <module>
    from tools import train
  File "/media/ailab/data/yy/CLUDA/tools/train.py", line 21, in <module>
    from mmseg.apis import set_random_seed, train_segmentor
  File "/media/ailab/data/yy/CLUDA/mmseg/apis/__init__.py", line 1, in <module>
    from .inference import inference_segmentor, init_segmentor, show_result_pyplot
  File "/media/ailab/data/yy/CLUDA/mmseg/apis/inference.py", line 12, in <module>
    from mmseg.models import build_segmentor
  File "/media/ailab/data/yy/CLUDA/mmseg/models/__init__.py", line 8, in <module>
    from .uda import *  # noqa: F401,F403
  File "/media/ailab/data/yy/CLUDA/mmseg/models/uda/__init__.py", line 2, in <module>
    from mmseg.models.uda.dacs import DACS
  File "/media/ailab/data/yy/CLUDA/mmseg/models/uda/dacs.py", line 31, in <module>
    from ..losses.contrastive_loss import contrastive_loss, centroid_cl, \
ImportError: cannot import name 'centroid_cl' from 'mmseg.models.losses.contrastive_loss' (/media/ailab/data/yy/CLUDA/mmseg/models/losses/contrastive_loss.py)

I find there is no centroid_cl in mmseg.models.losses.contrastive_loss.py