def assemble_masks(self, path):
# mask = np.zeros((self.config.IMG_HEIGHT, self.config.IMG_WIDTH), dtype=np.uint8)
mask = None
gauth = GoogleAuth()
gauth.LocalWebserverAuth() # Creates local webserver and auto handles authentication.
drive = GoogleDrive(gauth)
for i, mask_file in enumerate(next(os.walk(os.path.join(path, 'masks')))[2]):
#mask_ = Image.open(os.path.join(path, 'masks', mask_file)).convert("RGB")
mask_ = drive.open(os.path.join(path, 'masks', mask_file),'r')
# mask_ = mask_.resize((self.config.IMG_HEIGHT, self.config.IMG_WIDTH))
mask_ = np.asarray(mask_)
if i == 0:
mask = mask_
continue
mask = mask | mask_
# mask = np.expand_dims(mask, axis=-1)
return mask
