| 228 | |
| 229 | |
| 230 | class Assembler: |
| 231 | def __init__( |
| 232 | self, |
| 233 | data, |
| 234 | *, |
| 235 | max_n_individuals, |
| 236 | n_multibodyparts, |
| 237 | graph=None, |
| 238 | paf_inds=None, |
| 239 | greedy=False, |
| 240 | pcutoff=0.1, |
| 241 | min_affinity=0.05, |
| 242 | min_n_links=2, |
| 243 | max_overlap=0.8, |
| 244 | identity_only=False, |
| 245 | nan_policy="little", |
| 246 | force_fusion=False, |
| 247 | add_discarded=False, |
| 248 | window_size=0, |
| 249 | method="m1", |
| 250 | ): |
| 251 | self.data = data |
| 252 | self.metadata = self.parse_metadata(self.data) |
| 253 | self.max_n_individuals = max_n_individuals |
| 254 | self.n_multibodyparts = n_multibodyparts |
| 255 | self.n_uniquebodyparts = self.n_keypoints - n_multibodyparts |
| 256 | self.greedy = greedy |
| 257 | self.pcutoff = pcutoff |
| 258 | self.min_affinity = min_affinity |
| 259 | self.min_n_links = min_n_links |
| 260 | self.max_overlap = max_overlap |
| 261 | self._has_identity = "identity" in self[0] |
| 262 | if identity_only and not self._has_identity: |
| 263 | warnings.warn("The network was not trained with identity; setting `identity_only` to False.", stacklevel=2) |
| 264 | self.identity_only = identity_only & self._has_identity |
| 265 | self.nan_policy = nan_policy |
| 266 | self.force_fusion = force_fusion |
| 267 | self.add_discarded = add_discarded |
| 268 | self.window_size = window_size |
| 269 | self.method = method |
| 270 | self.graph = graph or self.metadata["paf_graph"] |
| 271 | self.paf_inds = paf_inds or self.metadata["paf"] |
| 272 | self._gamma = 0.01 |
| 273 | self._trees = dict() |
| 274 | self.safe_edge = False |
| 275 | self._kde = None |
| 276 | self.assemblies = dict() |
| 277 | self.unique = dict() |
| 278 | |
| 279 | def __getitem__(self, item): |
| 280 | return self.data[self.metadata["imnames"][item]] |
| 281 | |
| 282 | @classmethod |
| 283 | def empty( |
| 284 | cls, |
| 285 | max_n_individuals, |
| 286 | n_multibodyparts, |
| 287 | n_uniquebodyparts, |
no outgoing calls
no test coverage detected