1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
import os
from os.path import join
import logging
from app.settings import app_cfg as cfg
from app.settings import types
# -------------------------------------------------------------------------
# Metadata and media files
# -------------------------------------------------------------------------
class DataStore:
# local data store
def __init__(self, opt_data_store, opt_dataset):
self.data_store = join(f'/data_store_{opt_data_store.name.lower()}')
self.dir_dataset = join(self.data_store, 'datasets', cfg.DIR_PEOPLE, opt_dataset.name.lower())
self.dir_media = join(self.dir_dataset, 'media')
self.dir_metadata = join(self.dir_dataset, 'metadata')
def metadata(self, enum_type):
return join(self.dir_metadata, f'{enum_type.name.lower()}.csv')
def face_image(self, subdir, fn, ext):
return join(self.dir_media, 'original', subdir, f'{fn}.{ext}')
def face_image_crop(self, subdir, fn, ext):
return join(self.dir_media, 'cropped', subdir, f'{fn}.{ext}')
class DataStoreS3:
# S3 server
def __init__(self, opt_dataset):
self._dir_media = join(cfg.S3_HTTP_MEDIA_URL, opt_dataset.name.lower())
self._dir_metadata = join(cfg.S3_HTTP_METADATA_URL, opt_dataset.name.lower())
def metadata(self, opt_metadata_type, ext='csv'):
return join(self._dir_metadata, f'{opt_metadata_type.name.lower()}.{ext}')
def face_image(self, opt_uuid, ext='jpg'):
#return join(self._dir_media, 'original', f'{opt_uuid}.{ext}')
return join(self._dir_media, f'{opt_uuid}.{ext}')
def face_image_crop(self, opt_uuid, ext='jpg'):
# not currently using?
return join(self._dir_media, 'cropped', f'{opt_uuid}.{ext}')
# -------------------------------------------------------------------------
# Models
# -------------------------------------------------------------------------
|
