update README

This commit is contained in:
Qing
2022-09-04 21:23:58 +08:00
parent 2119a5f905
commit 3cc9521221
2 changed files with 21 additions and 6 deletions

View File

@@ -1161,6 +1161,9 @@ class FcF(InpaintModel):
masks: [H, W]
return: BGR IMAGE
"""
if image.shape[0] == 512 and image.shape[1] == 512:
return self._pad_forward(image, mask, config)
boxes = boxes_from_mask(mask)
crop_result = []
config.hd_strategy_crop_margin = 128