()
| 162 | |
| 163 | |
| 164 | def collect_model_downloads() -> Tuple[DownloadSet, DownloadSet]: |
| 165 | model_set = create_static_model_set('full') |
| 166 | model_hash_set = {} |
| 167 | model_source_set = {} |
| 168 | |
| 169 | for face_occluder_model in [ 'xseg_1', 'xseg_2', 'xseg_3' ]: |
| 170 | if state_manager.get_item('face_occluder_model') in [ 'many', face_occluder_model ]: |
| 171 | model_hash_set[face_occluder_model] = model_set.get(face_occluder_model).get('hashes').get('face_occluder') |
| 172 | model_source_set[face_occluder_model] = model_set.get(face_occluder_model).get('sources').get('face_occluder') |
| 173 | |
| 174 | for face_parser_model in [ 'bisenet_resnet_18', 'bisenet_resnet_34' ]: |
| 175 | if state_manager.get_item('face_parser_model') == face_parser_model: |
| 176 | model_hash_set[face_parser_model] = model_set.get(face_parser_model).get('hashes').get('face_parser') |
| 177 | model_source_set[face_parser_model] = model_set.get(face_parser_model).get('sources').get('face_parser') |
| 178 | |
| 179 | return model_hash_set, model_source_set |
| 180 | |
| 181 | |
| 182 | def pre_check() -> bool: |
no test coverage detected