Coverage for src/bob/bio/face/config/baseline/efficient_net.py: 100%

1from bob.bio.face.embeddings.pytorch import EfficientNet 

2from bob.bio.face.utils import lookup_config_from_database 

3 

4( 

5 annotation_type, 

6 fixed_positions, 

7 memory_demanding, 

8) = lookup_config_from_database(locals().get("database")) 

9 

10 

11def load(annotation_type, fixed_positions=None, memory_demanding=False): 

12 return EfficientNet(annotation_type, fixed_positions, memory_demanding) 

13 

14 

15pipeline = load(annotation_type, fixed_positions, memory_demanding) 

16 

17transformer = pipeline.transformer