15 lines
600 B
Python
15 lines
600 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
|
|
# ... 其他导入
|
|
|
|
PROCESSOR_MAP = {
|
|
"checkpoint": CheckpointWorker,
|
|
"trajectory": TrajectoryWorker,
|
|
"supervision_room": SupervisionWorker,
|
|
}
|
|
|
|
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] |