File size: 1,994 Bytes
ff66cf3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
from cliport.models.resnet import ResNet43_8s
from cliport.models.clip_wo_skip import CLIPWithoutSkipConnections
from cliport.models.rn50_bert_unet import RN50BertUNet
from cliport.models.rn50_bert_lingunet import RN50BertLingUNet
from cliport.models.rn50_bert_lingunet_lat import RN50BertLingUNetLat
from cliport.models.untrained_rn50_bert_lingunet import UntrainedRN50BertLingUNet
from cliport.models.clip_unet import CLIPUNet
from cliport.models.clip_lingunet import CLIPLingUNet
from cliport.models.resnet_lang import ResNet43_8s_lang
from cliport.models.resnet_lat import ResNet45_10s
from cliport.models.clip_unet_lat import CLIPUNetLat
from cliport.models.clip_lingunet_lat import CLIPLingUNetLat
from cliport.models.clip_film_lingunet_lat import CLIPFilmLingUNet
from cliport.models.clip_ling import CLIPLing
from cliport.models.resnet_lat_reduce import ResNet45_Reduced_10s
from cliport.models.pretrain_resnet import PretrainedResNet18
from cliport.models.mdetr_lingunet_lat_fuse import MDETRLingUNetLat_fuse
from cliport.models.resnet_lat_origin import ResNet45_10s_origin
names = {
# resnet
'plain_resnet': ResNet43_8s,
'plain_resnet_lang': ResNet43_8s_lang,
'pretrained_resnet18': PretrainedResNet18,
# without skip-connections
'clip_woskip': CLIPWithoutSkipConnections,
# unet
'clip_unet': CLIPUNet,
'rn50_bert_unet': RN50BertUNet,
# lingunet
'clip_lingunet': CLIPLingUNet,
'rn50_bert_lingunet': RN50BertLingUNet,
'untrained_rn50_bert_lingunet': UntrainedRN50BertLingUNet,
'clip_ling': CLIPLing,
# mdetr
'mdetr_lingunet_lat_fuse': MDETRLingUNetLat_fuse,
# lateral connections
'plain_resnet_lat': ResNet45_10s,
'plain_resnet_lat_origin': ResNet45_10s_origin,
'clip_unet_lat': CLIPUNetLat,
'clip_lingunet_lat': CLIPLingUNetLat,
'clip_film_lingunet_lat': CLIPFilmLingUNet,
'rn50_bert_lingunet_lat': RN50BertLingUNetLat,
'plain_resnet_reduce_lat': ResNet45_Reduced_10s,
}
|