apply lint
This commit is contained in:
parent
3f00da9681
commit
f4d944acbe
|
@ -71,7 +71,9 @@ def blend_tiles(
|
||||||
overlap: float,
|
overlap: float,
|
||||||
):
|
):
|
||||||
adj_tile = int(float(tile) * (1.0 - overlap))
|
adj_tile = int(float(tile) * (1.0 - overlap))
|
||||||
logger.trace("adjusting tile size from %s to %s based on %s overlap", tile, adj_tile, overlap)
|
logger.trace(
|
||||||
|
"adjusting tile size from %s to %s based on %s overlap", tile, adj_tile, overlap
|
||||||
|
)
|
||||||
|
|
||||||
scaled_size = (height * scale, width * scale, 3)
|
scaled_size = (height * scale, width * scale, 3)
|
||||||
count = np.zeros(scaled_size)
|
count = np.zeros(scaled_size)
|
||||||
|
@ -100,7 +102,13 @@ def blend_tiles(
|
||||||
# equalized size may be wrong/too much
|
# equalized size may be wrong/too much
|
||||||
scaled_bottom = min(scaled_top + equalized.shape[0], scaled_size[0])
|
scaled_bottom = min(scaled_top + equalized.shape[0], scaled_size[0])
|
||||||
scaled_right = min(scaled_left + equalized.shape[1], scaled_size[1])
|
scaled_right = min(scaled_left + equalized.shape[1], scaled_size[1])
|
||||||
logger.trace("tile broadcast shapes: %s, %s, %s, %s", scaled_top, scaled_bottom, scaled_left, scaled_right)
|
logger.trace(
|
||||||
|
"tile broadcast shapes: %s, %s, %s, %s",
|
||||||
|
scaled_top,
|
||||||
|
scaled_bottom,
|
||||||
|
scaled_left,
|
||||||
|
scaled_right,
|
||||||
|
)
|
||||||
|
|
||||||
# accumulation
|
# accumulation
|
||||||
value[scaled_top:scaled_bottom, scaled_left:scaled_right, :] += equalized[
|
value[scaled_top:scaled_bottom, scaled_left:scaled_right, :] += equalized[
|
||||||
|
|
|
@ -46,7 +46,6 @@ class ConversionContext(ServerContext):
|
||||||
reload: bool = True,
|
reload: bool = True,
|
||||||
share_unet: bool = True,
|
share_unet: bool = True,
|
||||||
extract: bool = False,
|
extract: bool = False,
|
||||||
|
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> None:
|
) -> None:
|
||||||
super().__init__(model_path=model_path, cache_path=cache_path, **kwargs)
|
super().__init__(model_path=model_path, cache_path=cache_path, **kwargs)
|
||||||
|
|
Loading…
Reference in New Issue