def upload_para_gdrive(self, caminho): """ Este método faz o upload de um arquivo para o Google Drive. Para que funcione, siga os passos a seguir: a. Entre no Google Developers Console <https://console.developers.google.com>, com a sua conta no Google, crie um novo projeto e habilite o Google Drive. b. Após configurar o projeto e tê-lo criado, entre em 'Credentials', selecione o projeto recém criado, e em 'Download JSON'. c. Renomeie o arquivo baixado para 'client_secrets.json' e coloque-o no mesmo diretório deste script. d. Ao ser executado, este script usará os parâmetros definidos no arquivo JSON para autenticar essa aplicação; em seguida, será aberto um navegador, para que o usuário acesse sua conta no Google e dê permissões de acesso a esta aplicação. ARGS: - path (string): caminho do arquivo a ser 'subido' para o Google Drive. """ f = GoogleDrive(GoogleAuth()).CreateFile() f.SetContentFile(caminho) f.Upload()
def prepare_training_data(self): # get imageId train_ids = next(os.walk(self.stage1_train_src))[1] gauth = GoogleAuth() drive = GoogleDrive(gauth) # read training data X_train = [] Y_train = [] print('reading training data starts...') sys.stdout.flush() for n, id_ in tqdm(enumerate(train_ids)): path = os.path.join(self.stage1_train_src,id_) dest = os.path.join(self.stage1_train_dest, id_) #img = Image.open(os.path.join(path, 'images', id_ + '.png'),'r').convert("RGB") fName= os.path.join(path, 'images', id_ + '.png') print(fName) img = drive.SetContentFile(fName) img.Upload() mask = self.assemble_masks(path) if not os.path.exists(dest): os.makedirs(dest) img.save(os.path.join(dest, 'image.png')) Image.fromarray(mask).save(os.path.join(dest, 'mask.png'))