(self, database, retriever_version='ViT-L/14')
| 60 | |
| 61 | class Searcher(object): |
| 62 | def __init__(self, database, retriever_version='ViT-L/14'): |
| 63 | assert database in DATABASES |
| 64 | # self.database = self.load_database(database) |
| 65 | self.database_name = database |
| 66 | self.searcher_savedir = f'data/rdm/searchers/{self.database_name}' |
| 67 | self.database_path = f'data/rdm/retrieval_databases/{self.database_name}' |
| 68 | self.retriever = self.load_retriever(version=retriever_version) |
| 69 | self.database = {'embedding': [], |
| 70 | 'img_id': [], |
| 71 | 'patch_coords': []} |
| 72 | self.load_database() |
| 73 | self.load_searcher() |
| 74 | |
| 75 | def train_searcher(self, k, |
| 76 | metric='dot_product', |
nothing calls this directly
no test coverage detected