from .conv_backbone import convnext_3d_small, convnext_3d_tiny | |
from .evaluator import COVER, BaseEvaluator, BaseImageEvaluator | |
from .head import IQAHead, VARHead, VQAHead | |
from .swin_backbone import SwinTransformer2D as IQABackbone | |
from .swin_backbone import SwinTransformer3D as VQABackbone | |
from .swin_backbone import swin_3d_small, swin_3d_tiny | |
__all__ = [ | |
"VQABackbone", | |
"IQABackbone", | |
"VQAHead", | |
"IQAHead", | |
"VARHead", | |
"BaseEvaluator", | |
"BaseImageEvaluator", | |
"COVER", | |
] | |