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'))
