Visitar URL original
Fix SkeletonBuilder initialization from GUI by C-Achard · Pull Request #3380 · DeepLabCut/DeepLabCut · GitHub
Skip to content
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion deeplabcut/gui/tabs/create_videos.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,4 +287,8 @@ def create_videos(self):
def build_skeleton(self, *args):
from deeplabcut.gui.widgets import SkeletonBuilder

SkeletonBuilder(self.root.config)
self.skeleton_builder = SkeletonBuilder(
config_path=self.root.config,
parent=self.root,
)
self.skeleton_builder.show()
9 changes: 6 additions & 3 deletions deeplabcut/gui/tabs/label_frames.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,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
from deeplabcut.gui.widgets import SkeletonBuilder, launch_napari


def label_frames(config_path: str | Path | None = None, image_folder: str | None = None):
Expand Down Expand Up @@ -140,4 +139,8 @@ def check_labels(self):
_ = launch_napari(labeled_images, plugin="napari", stack=True)

def build_skeleton(self, *args):
SkeletonBuilder(self.root.config)
self.skeleton_builder = SkeletonBuilder(
config_path=self.root.config,
parent=self.root,
)
self.skeleton_builder.show()
6 changes: 3 additions & 3 deletions deeplabcut/gui/widgets.py
Original file line number Diff line number Diff line change
Expand Up @@ -522,9 +522,9 @@ def display_help(self, *args):
)


class SkeletonBuilder(QtWidgets.QDialog, BaseSkeletonBuilder):
def __init__(self, config_path, parent=None):
QtWidgets.QDialog.__init__(self, parent)
class SkeletonBuilder(BaseSkeletonBuilder, QtWidgets.QDialog):
def __init__(self, *, config_path, parent=None):
QtWidgets.QDialog.__init__(self, parent=parent)
self._parent = parent
Comment thread
C-Achard marked this conversation as resolved.
Resolved
self.setWindowTitle("Skeleton Builder")
BaseSkeletonBuilder.__init__(self, config_path)
Expand Down
Loading