Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion basicsr/models/base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from copy import deepcopy
from torch.nn.parallel import DataParallel, DistributedDataParallel

from basicsr.models import lr_scheduler as lr_scheduler
#from basicsr.models import lr_scheduler as lr_scheduler
from basicsr.utils.dist_util import master_only

logger = logging.getLogger('basicsr')
Expand Down
4 changes: 4 additions & 0 deletions inference_codeformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -259,6 +259,10 @@ def set_realesrgan():
img_list = sorted(glob.glob(os.path.join(result_root, 'final_results', '*.[jp][pn]g')))
for img_path in img_list:
img = cv2.imread(img_path)
height, width = img.shape[:2]
wb = width % 2
hb = height % 2
img = cv2.copyMakeBorder(img, 0, hb, wb, 0, cv2.BORDER_REFLECT)
video_frames.append(img)
# write images to video
height, width = video_frames[0].shape[:2]
Expand Down