Skip to content
This repository was archived by the owner on Aug 28, 2023. It is now read-only.

Commit 5b7db5a

Browse files
authored
[65499] Return models to OMZ model list & Remove aclnet-int8 (#140)
1 parent 668f639 commit 5b7db5a

File tree

2 files changed

+6
-20
lines changed

2 files changed

+6
-20
lines changed

client/package-lock.json

+3-3
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

wb/main/console_tool_wrapper/model_downloader/utils.py

+3-17
Original file line numberDiff line numberDiff line change
@@ -485,6 +485,7 @@ def is_model_available(model: OMZTopologyModel) -> bool:
485485
def aggregate_topologies():
486486
# Filter rule #1: Show only models with not generic type
487487
not_supported_topologies = ('aclnet',
488+
'aclnet-int8',
488489
'asl-recognition-0004',
489490
'common-sign-language-0001',
490491
'common-sign-language-0002',
@@ -493,23 +494,8 @@ def aggregate_topologies():
493494
'mixnet-l',
494495
'netvlad-tf',
495496
'open-closed-eye-0001',
496-
# TODO: unskip when 65499 will be fixed
497-
'densenet-161-tf',
498-
'se-resnet-101',
499-
'se-resnet-50',
500-
'se-resnet-152',
501-
'drn-d-38',
502-
'se-inception',
503-
'densenet-169',
504-
'se-resnext-50',
505-
'ssd512',
506-
'Sphereface',
507-
'densenet-121',
508-
'densenet-161',
509-
'ssd300',
510-
'densenet-201',
511-
'se-resnext-101',
512-
'mobilenet-ssd',
497+
# 'drn-d-38' is not supported as there is no 'encode_segmentation_mask' in its accuracy configuration
498+
'drn-d-38'
513499
)
514500

515501
supported_topologies = OMZTopologyModel.query \

0 commit comments

Comments
 (0)