|
| 1 | +import argparse |
| 2 | +import os |
| 3 | +import os.path as osp |
| 4 | + |
| 5 | +import oss2 |
| 6 | + |
| 7 | +ACCESS_KEY_ID = os.getenv('OSS_ACCESS_KEY_ID', None) |
| 8 | +ACCESS_KEY_SECRET = os.getenv('OSS_ACCESS_KEY_SECRET', None) |
| 9 | +BUCKET_NAME = 'openmmlab' |
| 10 | +ENDPOINT = 'https://oss-accelerate.aliyuncs.com' |
| 11 | + |
| 12 | + |
| 13 | +def parse_args(): |
| 14 | + parser = argparse.ArgumentParser(description='Upload models to OSS') |
| 15 | + parser.add_argument('model_zoo', type=str, help='model_zoo input') |
| 16 | + parser.add_argument( |
| 17 | + '--dst-folder', |
| 18 | + type=str, |
| 19 | + default='mmsegmentation/v0.5', |
| 20 | + help='destination folder') |
| 21 | + args = parser.parse_args() |
| 22 | + return args |
| 23 | + |
| 24 | + |
| 25 | +def main(): |
| 26 | + args = parse_args() |
| 27 | + model_zoo = args.model_zoo |
| 28 | + dst_folder = args.dst_folder |
| 29 | + bucket = oss2.Bucket( |
| 30 | + oss2.Auth(ACCESS_KEY_ID, ACCESS_KEY_SECRET), ENDPOINT, BUCKET_NAME) |
| 31 | + |
| 32 | + for root, dirs, files in os.walk(model_zoo): |
| 33 | + for file in files: |
| 34 | + file_path = osp.relpath(osp.join(root, file), model_zoo) |
| 35 | + print(f'Uploading {file_path}') |
| 36 | + |
| 37 | + oss2.resumable_upload(bucket, osp.join(dst_folder, file_path), |
| 38 | + osp.join(model_zoo, file_path)) |
| 39 | + bucket.put_object_acl( |
| 40 | + osp.join(dst_folder, file_path), oss2.OBJECT_ACL_PUBLIC_READ) |
| 41 | + |
| 42 | + |
| 43 | +if __name__ == '__main__': |
| 44 | + main() |
0 commit comments