-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
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
Labels
No labels