Expand Up @@ -15,6 +15,7 @@ from deeplabcut.generate_training_dataset import check_labels from deeplabcut.gui.components import DefaultTab from deeplabcut.gui.widgets import launch_napari from deeplabcut.utils.skeleton import SkeletonBuilder

def label_frames(config_path): Expand All @@ -35,8 +36,11 @@ def _set_page(self): self.label_frames_btn.clicked.connect(self.label_frames) self.check_labels_btn = QtWidgets.QPushButton("Check Labels") self.check_labels_btn.clicked.connect(self.check_labels) self.build_skeleton_btn = QtWidgets.QPushButton("Build skeleton") self.build_skeleton_btn.clicked.connect(self.build_skeleton) self.main_layout.addWidget(self.label_frames_btn, alignment=Qt.AlignLeft) self.main_layout.addWidget(self.check_labels_btn, alignment=Qt.AlignLeft) self.main_layout.addWidget(self.build_skeleton_btn, alignment=Qt.AlignLeft)
def log_color_by_option(self, choice): self.root.logger.info(f"Labeled images will by colored by {choice.upper()}") Expand All @@ -63,3 +67,6 @@ def check_labels(self): check_labels(self.root.config, visualizeindividuals=self.root.is_multianimal) labeled_images = (Path(self.root.config).parent / "labeled-data").rglob("*_labeled/*.png") _ = launch_napari(labeled_images, plugin="napari", stack=True)
def build_skeleton(self, *args): SkeletonBuilder(self.root.config)