Блог

Я попытался загрузить изображение и сделать его доступным для модели.

 cracks_guess = []

    def check(arr_img):
        print('length->',len(arr_img))
        with torch.no_grad():
            for idx,img in enumerate(arr_img):
                img = torch.tensor(img).type(torch.FloatTensor).cuda()
                img = img.permute(2,0,1)
                img = img.unsqueeze(0)
                #print('shape->',img.shape)
                output = net(img)
                output = torch.sigmoid(output)
                pt = torch.round(output)
                if pt==1:
                #if output > 0.3:
                    cracks_guess.append(img)
                    print(idx,'output->',output)
 

Является ли это правильным способом изменить форму img на приемлемую для модели?