衛星データ+pytorch

以下のレポジトリ
github.com

Main_CarcountingA.ipynbの修正点

# データローダーからミニバッチを取り出すループ
        for inputs, labels in tqdm(dataloaders_dict[phase]):
          → num_labels = list(map(lambda x: int(x.replace('/','')), labels))
          → labels=torch.tensor(num_labels)
                
            # GPUが使えるならGPUにデータを送る
            inputs = inputs.to(device)
            labels = labels.to(device)