Spaces:
Runtime error
Runtime error
| import imgui | |
| from gui_utils import imgui_utils | |
| # ---------------------------------------------------------------------------- | |
| class ClassWidget: | |
| def __init__(self, viz): | |
| self.viz = viz | |
| self.cls = 0 | |
| self.animate = False | |
| self.count = 0 | |
| def __call__(self, show=True): | |
| viz = self.viz | |
| cls = self.cls | |
| if show: | |
| imgui.text('Class') | |
| imgui.same_line(viz.label_w) | |
| with imgui_utils.grayed_out(not viz.result.get('is_conditional', False)): | |
| _changed, self.cls = imgui.slider_int('##cls', self.cls, 0, viz.result.get('num_classes', 0) - 1) | |
| imgui.same_line() | |
| _clicked, self.animate = imgui.checkbox('Anim##cls', self.animate) | |
| imgui.same_line() | |
| if imgui_utils.button('Reset', width=-1, enabled=(cls != self.cls or self.animate)): | |
| self.cls = 0 | |
| self.animate = False | |
| self.count = 0 | |
| if self.animate: | |
| self.count += self.viz.frame_delta | |
| if self.count > 1.5: # Update the class every 1.5 seconds; arbitrary, change as you will | |
| self.cls = (self.cls + 1) % viz.result.get('num_classes') # Loop back | |
| self.count = 0 | |
| # Sanity check when loading new networks | |
| self.cls = min(self.cls, viz.result.get('num_classes', 1) - 1) | |
| viz.args.update(cls=self.cls) | |