f26ee72ff530d6d4ef552de13f8fdeeeac0e4a18,dvalib/indexer.py,BaseIndexer,apply,#BaseIndexer#Any#,20

Before Change

    def apply(self,path):
        self.load()
        tensor = self.transform(PIL.Image.open(path).convert("RGB"))
        return self.net(Variable(tensor.unsqueeze_(0))).data.numpy()

    def load_index(self,path):
        temp_index = []

After Change

    def apply(self,path):
        self.load()
        tensor = self.transform(PIL.Image.open(path).convert("RGB")).unsqueeze_(0)
        if torch.cuda.is_available():
            tensor = tensor.cuda()
        return self.net(Variable(tensor)).data.numpy()

    def load_index(self,path):
        temp_index = []

Instances