Shufflenet v2 pretrained (#277)

* add two more pretrained models for shufflenet * fix unzip single file
Oneflow-Inc · Nov 2, 2023 · af65737 · af65737
1 parent adab8ca
commit af65737
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/flowvision/models/shufflenet_v2.py b/flowvision/models/shufflenet_v2.py
@@ -22,8 +22,8 @@
 model_urls = {
  "shufflenet_v2_x0_5": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/ShuffleNetV2/shufflenet_v2_x0_5.zip",
  "shufflenet_v2_x1_0": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/ShuffleNetV2/shufflenet_v2_x1_0.zip",
- "shufflenet_v2_x1_5": None,
- "shufflenet_v2_x2_0": None,
+ "shufflenet_v2_x1_5": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/ShuffleNetV2/shufflenet_v2_x1_5.zip",
+ "shufflenet_v2_x2_0": "https://oneflow-public.oss-cn-beijing.aliyuncs.com/model_zoo/flowvision/classification/ShuffleNetV2/shufflenet_v2_x2_0.zip",
 }
 
 

diff --git a/flowvision/models/utils.py b/flowvision/models/utils.py
@@ -73,8 +73,6 @@ def _legacy_zip_load(filename, model_dir, map_location, delete_zip_file=True):
  members = f.infolist()
  extracted_name = members[0].filename
  extracted_file = os.path.join(model_dir, extracted_name)
- if not os.path.exists(extracted_file):
- os.mkdir(extracted_file)
  f.extractall(model_dir)
  if delete_zip_file and os.path.exists(filename):
  os.remove(filename)