(
cls,
max_n_individuals,
n_multibodyparts,
n_uniquebodyparts,
graph,
paf_inds,
greedy=False,
pcutoff=0.1,
min_affinity=0.05,
min_n_links=2,
max_overlap=0.8,
identity_only=False,
nan_policy="little",
force_fusion=False,
add_discarded=False,
window_size=0,
method="m1",
)
| 281 | |
| 282 | @classmethod |
| 283 | def empty( |
| 284 | cls, |
| 285 | max_n_individuals, |
| 286 | n_multibodyparts, |
| 287 | n_uniquebodyparts, |
| 288 | graph, |
| 289 | paf_inds, |
| 290 | greedy=False, |
| 291 | pcutoff=0.1, |
| 292 | min_affinity=0.05, |
| 293 | min_n_links=2, |
| 294 | max_overlap=0.8, |
| 295 | identity_only=False, |
| 296 | nan_policy="little", |
| 297 | force_fusion=False, |
| 298 | add_discarded=False, |
| 299 | window_size=0, |
| 300 | method="m1", |
| 301 | ): |
| 302 | # Dummy data |
| 303 | n_bodyparts = n_multibodyparts + n_uniquebodyparts |
| 304 | data = { |
| 305 | "metadata": { |
| 306 | "all_joints_names": ["" for _ in range(n_bodyparts)], |
| 307 | "PAFgraph": graph, |
| 308 | "PAFinds": paf_inds, |
| 309 | }, |
| 310 | "0": {}, |
| 311 | } |
| 312 | return cls( |
| 313 | data, |
| 314 | max_n_individuals=max_n_individuals, |
| 315 | n_multibodyparts=n_multibodyparts, |
| 316 | graph=graph, |
| 317 | paf_inds=paf_inds, |
| 318 | greedy=greedy, |
| 319 | pcutoff=pcutoff, |
| 320 | min_affinity=min_affinity, |
| 321 | min_n_links=min_n_links, |
| 322 | max_overlap=max_overlap, |
| 323 | identity_only=identity_only, |
| 324 | nan_policy=nan_policy, |
| 325 | force_fusion=force_fusion, |
| 326 | add_discarded=add_discarded, |
| 327 | window_size=window_size, |
| 328 | method=method, |
| 329 | ) |
| 330 | |
| 331 | @property |
| 332 | def n_keypoints(self): |
no outgoing calls