mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +00:00
34 lines
823 B
Python
34 lines
823 B
Python
|
"""
|
||
|
|
Copyright (c) 2022, salesforce.com, inc.
|
||
|
|
All rights reserved.
|
||
|
|
SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause
|
||
|
|
"""
|
||
|
|
|
||
|
|
from minigpt4.processors.base_processor import BaseProcessor
|
||
|
|
from minigpt4.processors.blip_processors import (
|
||
|
|
Blip2ImageTrainProcessor,
|
||
|
|
Blip2ImageEvalProcessor,
|
||
|
|
BlipCaptionProcessor,
|
||
|
|
)
|
||
|
|
|
||
|
|
from minigpt4.common.registry import registry
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"BaseProcessor",
|
||
|
|
"Blip2ImageTrainProcessor",
|
||
|
|
"Blip2ImageEvalProcessor",
|
||
|
|
"BlipCaptionProcessor",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def load_processor(name, cfg=None):
|
||
|
|
"""
|
||
|
|
Example
|
||
|
|
|
||
|
|
>>> processor = load_processor("alpro_video_train", cfg=None)
|
||
|
|
"""
|
||
|
|
processor = registry.get_processor_class(name).from_config(cfg)
|
||
|
|
|
||
|
|
return processor
|