22 lines
887 B
Python
22 lines
887 B
Python
from biz.checkpoint.checkpoint_biz import FrameProcessorWorker as CheckpointWorker
|
|
from biz.prison.trajectory02_biz import FrameProcessorWorker as TrajectoryWorker
|
|
from biz.prison.supervision_room_biz import FrameProcessorWorker as SupervisionWorker
|
|
from biz.prison.ab_biz import FrameProcessorWorker as AbWorker
|
|
from biz.prison.prison_biz import FrameProcessorWorker as CorridorWorker
|
|
from biz.prison.indoor_biz import FrameProcessorWorker as IndoorWorker
|
|
|
|
# ... 其他导入
|
|
|
|
PROCESSOR_MAP = {
|
|
"checkpoint": CheckpointWorker,
|
|
"trajectory": TrajectoryWorker,
|
|
"supervision_room": SupervisionWorker,
|
|
"ab": AbWorker,
|
|
"corridor": CorridorWorker,
|
|
"indoor": IndoorWorker
|
|
}
|
|
|
|
def get_processor(processor_type: str):
|
|
if processor_type not in PROCESSOR_MAP:
|
|
raise ValueError(f"Unknown processor type: {processor_type}")
|
|
return PROCESSOR_MAP[processor_type] |