Skip to content

config_path #1

@stephenzhu1212

Description

@stephenzhu1212

UnifiedWatermarkEngine类中:
def _get_image_watermark(self) -> ImageWatermark:
"""获取图像水印处理器"""
if self._image_watermark is None:
self._image_watermark = ImageWatermark(self.config_path)
self._image_watermark.algorithm = 'videoseal'
return self._image_watermark

def _get_audio_watermark(self) -> AudioWatermark:
    """获取音频水印处理器"""
    if self._audio_watermark is None:
        self._audio_watermark = AudioWatermark(self.config_path)
    return self._audio_watermark

def _get_video_watermark(self) -> VideoWatermark:
    """获取视频水印处理器"""
    if self._video_watermark is None:
        from ..video_watermark.video_watermark import create_video_watermark
        self._video_watermark = create_video_watermark()
    return self._video_watermark

视频类型的config_path并不会向里面传递,而是在VideoSealWrapper的load_model方法中使用了写死的路径:
try:
# 使用videoseal的默认模型加载方式(指定模型卡)
self.model = videoseal.load("videoseal_1.0")
finally:
# 恢复工作目录
os.chdir(current_dir)

会导致传入的config_path不起作用!!!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions