Merge branch 'development' into asymmetric-tiling

This commit is contained in:
Carson Katri
2022-10-18 22:37:18 -04:00
committed by GitHub
2 changed files with 3 additions and 2 deletions

View File

@ -29,7 +29,7 @@ work fine.
import torch
import numpy as np
from models.clipseg import CLIPDensePredT
from clipseg import CLIPDensePredT
from einops import rearrange, repeat
from PIL import Image
from torchvision import transforms

View File

@ -117,7 +117,7 @@ try:
with zipfile.ZipFile(model_dest,'r') as zip:
zip.extractall('src/clipseg')
os.rename('src/clipseg/clipseg_weights','src/clipseg/weights')
from models.clipseg import CLIPDensePredT
from clipseg import CLIPDensePredT
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, )
model.eval()
model.load_state_dict(
@ -126,6 +126,7 @@ try:
map_location=torch.device('cpu'),
strict=False,
)
)
except Exception:
print('Error installing clipseg model:')
print(traceback.format_exc())