Ma/video-subtitle-remover
1
0
Fork 0
mirror of https://github.com/YaoFANGUK/video-subtitle-remover.git synced 2026-02-19 07:44:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update sttn_inpaint.py

Browse Source
This commit is contained in:
天涯古巷
2023-12-23 10:42:31 +08:00
committed by GitHub
parent a918517429
commit 5c819f70d1
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/inpaint/sttn_inpaint.py
View File

@@ -4,7 +4,9 @@ import numpy as np
import torch
from torchvision import transforms
from typing import List
import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
from backend import config
from backend.inpaint.sttn.auto_sttn import InpaintGenerator
from backend.inpaint.utils.sttn_utils import Stack, ToTorchFormatTensor