fix bug of ImageMaskScaleAs
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import torch
|
||||
from PIL import Image
|
||||
from .imagefunc import AnyType, log, tensor2pil, pil2tensor, fit_resize_image
|
||||
from .imagefunc import AnyType, log, tensor2pil, pil2tensor, image2mask, fit_resize_image
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user