5e82853bb435e7c46e2e3bafa2c1d0a4c40ffb38,dvalib/indexer.py,BaseIndexer,apply,#BaseIndexer#Any#,20

Before Change

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

    def load_index(self,path):
        temp_index = []
        for dirname in os.listdir(path +"/"):

After Change

        if torch.cuda.is_available():
            tensor = torch.FloatTensor(tensor).cuda()
        result = self.net(Variable(tensor))
        if torch.cuda.is_available():
            result = result.data.cpu()
        return result.data.numpy()

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

Instances