PramaLLC
/

BEN2

Image Segmentation

background-remove

mask-generation

Dichotomous image segmentation

background remove

remove background

model_hub_mixin

pytorch_model_hub_mixin

background removal

background-removal

Model card Files Files and versions

MaxwellMeyer commited on Feb 2

Commit

b210846

·

verified ·

1 Parent(s): 11be3ff

Update BEN2.py

Files changed (1) hide show

BEN2.py +1 -1

BEN2.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 from einops import rearrange
 import torch.utils.checkpoint as checkpoint
 import numpy as np
-from timm.layers import DropPath, to_2tuple, trunc_normal_
 from PIL import Image, ImageOps
 from torchvision import transforms
 import numpy as np

 from einops import rearrange
 import torch.utils.checkpoint as checkpoint
 import numpy as np
+from timm.models.layers  import DropPath, to_2tuple, trunc_normal_
 from PIL import Image, ImageOps
 from torchvision import transforms
 import numpy as np