(data_dict, text_feat_path, save_path)
| 214 | |
| 215 | |
| 216 | def visualize_3d(data_dict, text_feat_path, save_path): |
| 217 | text_feat = torch.load(text_feat_path) |
| 218 | group_logits = np.einsum('nc,mc->nm', data_dict["group_feat"], text_feat) |
| 219 | group_labels = np.argmax(group_logits, axis=-1) |
| 220 | labels = group_labels[data_dict["group"]] |
| 221 | labels[data_dict["group"] == -1] = -1 |
| 222 | visualize_pcd(data_dict["coord"], data_dict["color"], labels, save_path) |
| 223 | |
| 224 | |
| 225 | def visualize_pcd(coord, pcd_color, labels, save_path): |
nothing calls this directly
no test coverage detected