Skip to content

Commit f95079b

Browse files
authored
[Enhance] Migrate to MMCV DepthwiseSeparableConv (#158)
* Add D16-MG124 models * Use MMCV DepthSepConv * add OHEM * add warmup * fixed test * fixed test * change to bs 16 * revert config * add models * seperate
1 parent b9fd8e4 commit f95079b

14 files changed

+55
-556
lines changed

.dev/clean_models.py

-125
This file was deleted.

.dev/generate_table.py

-152
This file was deleted.

.dev/modelzoo_json2md.py

-58
This file was deleted.

.dev/upload_modelzoo.py

+44
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
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()

mmseg/core/utils/__init__.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from .dist_utils import allreduce_grads
21
from .misc import add_prefix
32

4-
__all__ = ['add_prefix', 'allreduce_grads']
3+
__all__ = ['add_prefix']

0 commit comments

Comments
 (0)