training/detectors/__init__.py

import os
import sys
current_file_path = os.path.abspath(__file__)
parent_dir = os.path.dirname(os.path.dirname(current_file_path))
project_root_dir = os.path.dirname(parent_dir)
sys.path.append(parent_dir)
sys.path.append(project_root_dir)

from metrics.registry import DETECTOR
from .utils import slowfast

from .facexray_detector import FaceXrayDetector
from .xception_detector import XceptionDetector
from .efficientnetb4_detector import EfficientDetector
from .resnet34_detector import ResnetDetector
from .f3net_detector import F3netDetector
from .meso4_detector import Meso4Detector
from .meso4Inception_detector import Meso4InceptionDetector
from .spsl_detector import SpslDetector
from .core_detector import CoreDetector
from .capsule_net_detector import CapsuleNetDetector
from .srm_detector import SRMDetector
from .ucf_detector import UCFDetector
from .recce_detector import RecceDetector
from .fwa_detector import FWADetector
from .ffd_detector import FFDDetector

from .clip_detector import CLIPDetector
from .timesformer_detector import TimeSformerDetector
from .xclip_detector import XCLIPDetector
from .sbi_detector import SBIDetector
from .ftcn_detector import FTCNDetector
from .i3d_detector import I3DDetector
from .altfreezing_detector import AltFreezingDetector
from .stil_detector import STILDetector
from .lsda_detector import LSDADetector
from .sladd_detector import SLADDXceptionDetector
from .pcl_xception_detector import PCLXceptionDetector
from .iid_detector import IIDDetector
from .lrl_detector import LRLDetector
from .rfm_detector import RFMDetector
from .uia_vit_detector import UIAViTDetector
from .multi_attention_detector import MultiAttentionDetector
from .sia_detector import SIADetector
from .tall_detector import TALLDetector
from .cnn_dct_detector import CNNDCTDetector
from .clip_image_detector import CLIPImageDetector
from .dino_contrast import DINOv2_Large_FFT_Contrast_Detector

from .d_det import DNADetector
from .universal import UniversalDetector
from .clip_large_detector import CLIP_Large_Detector
# Effort Models
from .effort_detector import EffortDetector
from .effort_vid_detector import EffortVidDetector # 16 frames + avg
# VideoMAE LoRA
from .videomae_detector import VideoMAEDetector
from .videomae_large_detector import VideoMAELargeDetector
from .videomae_lora_detector import VideoMAELoRADetector
# CLIP-ViT FFT
from .clip_large_fft_detector import CLIP_Large_FFT_Detector
from .clip_base_fft_detector import CLIP_Base_FFT_Detector
# CLIP-ViT LoRA r16
from .clip_base_vid_detector import CLIP_Base_Vid_Detector
from .clip_large_vid_detector import CLIP_Large_Vid_Detector
from .clip_openai_vid_detector import CLIP_Openai_Large_Vid_Detector
from .clip_large_lora_detector import CLIP_Large_LoRA_Detector
# CLIP-ViT-Adapter
from .clip_adapter_two_3dconv_detector import CLIPAdapter3DConvDetector

# CLIP Contrast
from .clip_contrast_detector import CLIP_Contrast
from .clip_hier_contrast_detector import CLIP_Contrast_HIER

#aug
from .clip_large_lsda import CLIP_Large_FFT_LSDA_Detector
from .clip_patch_shuffle import CLIP_PATCH_SHUFFLE_Detector
from .vit_detector import ViT_Large_FFT_Detector
from .effort_patch_shuffle import Effort_Shuffle_Ensenble_Detector


# DINO
from .dinov2_large_fft_detector import DINOv2_Large_FFT_Detector
from .dinov3_large_fft_detector import DINOv3_Large_FFT_Detector
from .clip_large_fft_supcon_detector import CLIP_Large_FFT_SupCon_Detector
# SupConCls
from .clip_large_fft_supcon_cls_detector import CLIP_Large_FFT_SupCon_Cls_Detector
# Dis
from .clip_large_fft_dis_detector import CLIP_Large_FFT_Dis_Detector
from .clip_large_fft_dis_orth_detector import CLIP_Large_FFT_Dis_Orth_Detector
from .clip_large_fft_dis_orth1_detector import CLIP_Large_FFT_Dis_Orth1_Detector # Full features -net> semantic features; full features - semantic features; orthogonal loss
from .clip_large_fft_dis_orth2_detector import CLIP_Large_FFT_Dis_Orth2_Detector # Full features -net> semantic features; subtract the projection of full features onto semantic features; orthogonal loss with dual mapping
from .clip_large_fft_dis_orth3_detector import CLIP_Large_FFT_Dis_Orth3_Detector # Full features -net> semantic features; subtract the projection of full features onto semantic features; orthogonal loss with dual mapping, using ReLU in the projection layer
# VAE
from .clip_large_fft_vae1_detector import CLIP_Large_FFT_VAE1_Detector
from .clip_large_fft_vae2_detector import CLIP_Large_FFT_VAE2_Detector
# Concat
from .clip_large_fft_dis_cat1_detector import CLIP_Large_FFT_Dis_Cat1_Detector
from .clip_large_fft_dis_cat2_detector import CLIP_Large_FFT_Dis_Cat2_Detector
from .clip_large_fft_dino_orth_detector import CLIP_Large_FFT_Dino_Orth_Detector
# Effort CL
from .effort_cl_detector import EffortCLDetector
# AE
from .ae_detector import LDM_AE_Classify_Detector
from .ae_detector_resnet34 import ResNet34_AE_Trace_Detector
# POSE
from .pose_detector import POSE_Detector
# HRNet
from .hrnet_detector import HRNet_Detector
# RepMix
from .repmix_detector import RepMix_Detector
# Lorax
from .lorax_detector import LoRAXConvitDetector
# GANAtt
from .ganatt_detector import GANAtt_Detector
# NPR
from .npr_detector import NPR
#OOC
from .ooc_detector import OOCDetector

from .resnet34_distill_detector import DetectorDistill