(self, idx)
| 84 | return len(self.ids) * self.k |
| 85 | |
| 86 | def __getitem__(self, idx): |
| 87 | src_id = self.ids[idx // self.k] |
| 88 | tar_id = random.choice(self.ids) |
| 89 | # positive sample for discriminator network |
| 90 | real_id = random.choice(self.ids) |
| 91 | src_name = self.models[src_id] |
| 92 | tar_name = self.models[tar_id] |
| 93 | src_pc = self.pc[src_id] |
| 94 | tar_pc = self.pc[tar_id] |
| 95 | key_pts = self.key_pts[src_id] |
| 96 | w_mesh = self.w_mesh[src_id] |
| 97 | w_pc = self.w_pc[src_id] |
| 98 | src_ver = self.mesh_vertices[src_id] |
| 99 | src_face = self.mesh_faces[src_id] |
| 100 | tar_ver = self.mesh_vertices[tar_id] |
| 101 | tar_face = self.mesh_faces[tar_id] |
| 102 | real_ver = self.mesh_vertices[real_id] |
| 103 | real_face = self.mesh_faces[real_id] |
| 104 | return {"src_name": src_name, "tar_name": tar_name, |
| 105 | "src_pc": src_pc, "tar_pc": tar_pc, |
| 106 | "src_ver": src_ver, "src_face": src_face, |
| 107 | "tar_ver": tar_ver, "tar_face": tar_face, |
| 108 | "real_ver": real_ver, "real_face": real_face, |
| 109 | "key_pts": key_pts, "w_mesh": w_mesh, "w_pc": w_pc} |
nothing calls this directly
no outgoing calls
no test coverage detected