AI-Cyber commited on
Commit
b08001d
·
1 Parent(s): ca459a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def Decoder1_optical_instance(image_input):
64
  img = transforms.Resize([1024, 1024])(image_input)
65
  transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229,0.224,0.225])])
66
  input_img = transform(img)
67
- input_img = transforms.ToTensor()(img).unsqueeze(0)
68
  image_embedding = model.image_encoder(input_img) # torch.Size([1, 256, 64, 64])
69
  sparse_embeddings, dense_embeddings, scatter_embeddings = model.prompt_encoder(
70
  points=None,
 
64
  img = transforms.Resize([1024, 1024])(image_input)
65
  transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229,0.224,0.225])])
66
  input_img = transform(img)
67
+ input_img = input_img.unsqueeze(0)
68
  image_embedding = model.image_encoder(input_img) # torch.Size([1, 256, 64, 64])
69
  sparse_embeddings, dense_embeddings, scatter_embeddings = model.prompt_encoder(
70
  points=None,