ISAT package

Subpackages

• ISAT.formats package
  • Submodules
  • ISAT.formats.coco module
    • COCO
      • COCO.keep_crowd
      • COCO.read_from_coco()
      • COCO.save_to_coco()
  • ISAT.formats.isat module
    • ISAT
      • ISAT.annos
      • ISAT.cates
      • ISAT.ANNO
      • ISAT.annos
      • ISAT.cates
      • ISAT.read_from_ISAT()
      • ISAT.save_to_ISAT()
      • ISAT.remove_file_suffix()
      • ISAT.load_one_isat_json()
      • ISAT.save_one_isat_json()
  • ISAT.formats.labelme module
    • LABELME
      • LABELME.keep_crowd
      • LABELME.read_from_LABELME()
      • LABELME.save_to_LABELME()
      • LABELME.save_one_labelme_json()
      • LABELME.load_one_labelme_json()
  • ISAT.formats.voc module
    • VOC
      • VOC.keep_crowd
      • VOC.is_instance
      • VOC.save_to_voc()
      • VOC.save_one_voc_png()
      • VOC.fill_polygon()
  • ISAT.formats.voc_detection module
    • VOCDetect
      • VOCDetect.keep_crowd
      • VOCDetect.save_to_XML()
      • VOCDetect.save_one_voc_xml()
  • ISAT.formats.yolo module
    • YOLO
      • YOLO.read_from_YOLO()
      • YOLO.save_to_YOLO()
      • YOLO.merge_multi_segment()
      • YOLO.yolo2isat_segmentation()
      • YOLO.get_isat_bbox()
      • YOLO.bbox_within()
      • YOLO.load_one_yolo_txt()
      • YOLO.save_one_yolo_txt()
  • Module contents
• ISAT.segment_any package
  • Submodules
  • ISAT.segment_any.gpu_resource module
    • GPUResource_Thread
      • GPUResource_Thread.message
      • GPUResource_Thread.run()
  • ISAT.segment_any.model_zoo module
  • ISAT.segment_any.segment_any module
    • SegAny
      • SegAny.checkpoint
      • SegAny.model_dtype
      • SegAny.model_source
      • SegAny.model_type
      • SegAny.predictor
      • SegAny.set_image()
      • SegAny.reset_image()
      • SegAny.predict_with_point_prompt()
      • SegAny.predict_with_box_prompt()
    • SegAnyVideo
      • SegAnyVideo.checkpoint
      • SegAnyVideo.model_dtype
      • SegAnyVideo.model_source
      • SegAnyVideo.model_type
      • SegAnyVideo.predictor
      • SegAnyVideo.init_state()
      • SegAnyVideo.reset_state()
      • SegAnyVideo.add_new_mask()
  • Module contents
• ISAT.utils package
  • Submodules
  • ISAT.utils.dicom module
    • load_dcm_as_image()
  • Module contents
• ISAT.widgets package
  • Submodules
  • ISAT.widgets.about_dialog module
    • AboutDialog
  • ISAT.widgets.annos_dock_widget module
    • AnnosDockWidget
      • AnnosDockWidget.mainwindow
      • AnnosDockWidget.polygon_item_dict
      • AnnosDockWidget.right_button_menu()
      • AnnosDockWidget.generate_item_and_itemwidget()
      • AnnosDockWidget.update_listwidget()
      • AnnosDockWidget.update_combobox_group_select()
      • AnnosDockWidget.listwidget_add_polygon()
      • AnnosDockWidget.listwidget_remove_polygon()
      • AnnosDockWidget.set_selected()
      • AnnosDockWidget.set_polygon_selected()
      • AnnosDockWidget.set_polygon_show()
      • AnnosDockWidget.set_all_polygon_visible()
      • AnnosDockWidget.set_all_polygon_lock()
      • AnnosDockWidget.set_group_polygon_visible()
      • AnnosDockWidget.zoom_to_group()
      • AnnosDockWidget.go_to_next_group()
      • AnnosDockWidget.go_to_prev_group()
  • ISAT.widgets.annos_validator_dialog module
    • AnnosValidatorDialog
      • AnnosValidatorDialog.open_dir()
      • AnnosValidatorDialog.validate()
  • ISAT.widgets.auto_segment_dialog module
    • AutoSegmentThread
      • AutoSegmentThread.message
      • AutoSegmentThread.run()
    • AutoSegmentDialog
      • AutoSegmentDialog.open_dir()
      • AutoSegmentDialog.start()
      • AutoSegmentDialog.cancel()
      • AutoSegmentDialog.print_message()
      • AutoSegmentDialog.init_connect()
  • ISAT.widgets.canvas module
    • AnnotationScene
      • AnnotationScene.image_item
      • AnnotationScene.mask_item
      • AnnotationScene.image_data
      • AnnotationScene.current_graph
      • AnnotationScene.prompt_box_item
      • AnnotationScene.repaint_line_item
      • AnnotationScene.mode
      • AnnotationScene.draw_mode
      • AnnotationScene.contour_mode
      • AnnotationScene.prompt_point_positions
      • AnnotationScene.prompt_point_labels
      • AnnotationScene.prompt_point_items
      • AnnotationScene.mask
      • AnnotationScene.mask_alpha
      • AnnotationScene.guide_line_x
      • AnnotationScene.guide_line_y
      • AnnotationScene.last_draw_time
      • AnnotationScene.draw_interval
      • AnnotationScene.pressed
      • AnnotationScene.selected_polygons_list
      • AnnotationScene.repaint_start_vertex
      • AnnotationScene.repaint_end_vertex
      • AnnotationScene.hovered_vertex
      • AnnotationScene.load_image()
      • AnnotationScene.unload_image()
      • AnnotationScene.change_mode_to_create()
      • AnnotationScene.change_mode_to_view()
      • AnnotationScene.change_mode_to_edit()
      • AnnotationScene.change_mode_to_repaint()
      • AnnotationScene.change_contour_mode_to_save_all()
      • AnnotationScene.change_contour_mode_to_save_max_only()
      • AnnotationScene.change_contour_mode_to_save_external()
      • AnnotationScene.change_contour_method_to_simple()
      • AnnotationScene.change_contour_method_to_tc89_kcos()
      • AnnotationScene.change_contour_method_to_none()
      • AnnotationScene.start_segment_anything()
      • AnnotationScene.start_segment_anything_box()
      • AnnotationScene.start_segment_anything_visual()
      • AnnotationScene.start_draw_polygon()
      • AnnotationScene.start_draw()
      • AnnotationScene.finish_draw()
      • AnnotationScene.cancel_draw()
      • AnnotationScene.delete_selected_graph()
      • AnnotationScene.edit_polygon()
      • AnnotationScene.move_polygon_to_top()
      • AnnotationScene.move_polygon_to_bottom()
      • AnnotationScene.copy_item()
      • AnnotationScene.polygons_union()
      • AnnotationScene.polygons_difference()
      • AnnotationScene.polygons_intersection()
      • AnnotationScene.polygons_symmetric_difference()
      • AnnotationScene.mousePressEvent()
      • AnnotationScene.mouseReleaseEvent()
      • AnnotationScene.mouseMoveEvent()
      • AnnotationScene.update_mask()
      • AnnotationScene.backspace()
    • AnnotationView
      • AnnotationView.keyPressEvent()
      • AnnotationView.keyReleaseEvent()
      • AnnotationView.wheelEvent()
      • AnnotationView.zoom_in()
      • AnnotationView.zoom_out()
      • AnnotationView.zoomfit()
      • AnnotationView.zoom()
      • AnnotationView.horizontal_scroll()
      • AnnotationView.vertical_scroll()
  • ISAT.widgets.category_dock_widget module
    • CategoriesDockWidget
      • CategoriesDockWidget.choice_category()
      • CategoriesDockWidget.update_widget()
      • CategoriesDockWidget.radio_choice()
      • CategoriesDockWidget.item_choice()
      • CategoriesDockWidget.update_current_group()
      • CategoriesDockWidget.increase_current_group()
      • CategoriesDockWidget.decrease_current_group()
      • CategoriesDockWidget.toggle_group_mode()
  • ISAT.widgets.category_edit_dialog module
    • CategoryEditDialog
      • CategoryEditDialog.check_category_enabled()
      • CategoryEditDialog.check_group_enabled()
      • CategoryEditDialog.check_note_enabled()
      • CategoryEditDialog.check_crowded_enabled()
      • CategoryEditDialog.load_cfg()
      • CategoryEditDialog.get_category()
      • CategoryEditDialog.apply()
      • CategoryEditDialog.cancel()
      • CategoryEditDialog.closeEvent()
      • CategoryEditDialog.reject()
  • ISAT.widgets.category_setting_dialog module
    • CategorySettingDialog
      • CategorySettingDialog.get_item_and_widget()
      • CategorySettingDialog.remove_category_item()
      • CategorySettingDialog.load_cfg()
      • CategorySettingDialog.add_new_category()
      • CategorySettingDialog.choice_color()
      • CategorySettingDialog.import_cfg()
      • CategorySettingDialog.export_cfg()
      • CategorySettingDialog.apply()
      • CategorySettingDialog.cancel()
      • CategorySettingDialog.init_connect()
  • ISAT.widgets.converter_dialog module
    • Converter
      • Converter.message
      • Converter.run()
      • Converter.load_from_isat()
      • Converter.convert_to_isat()
    • COCOConverter
      • COCOConverter.run()
      • COCOConverter.coco_to_isat()
      • COCOConverter.isat_to_coco()
    • YOLOConverter
      • YOLOConverter.run()
      • YOLOConverter.yolo_to_isat()
      • YOLOConverter.isat_to_yolo()
    • LABELMEConverter
      • LABELMEConverter.run()
      • LABELMEConverter.labelme_to_isat()
      • LABELMEConverter.isat_to_labelme()
    • VOCConverter
      • VOCConverter.run()
      • VOCConverter.isat_to_voc()
    • VOCConverterForDetection
      • VOCConverterForDetection.run()
      • VOCConverterForDetection.isat_to_voc_for_detection()
    • ConverterDialog
      • ConverterDialog.tab_widget_current_changed()
      • ConverterDialog.apply()
      • ConverterDialog.cancel()
      • ConverterDialog.open_file()
      • ConverterDialog.save_file()
      • ConverterDialog.open_dir()
      • ConverterDialog.save_dir()
      • ConverterDialog.print_message()
      • ConverterDialog.init_connect()
  • ISAT.widgets.files_dock_widget module
    • FilesDockWidget
      • FilesDockWidget.update_widget()
      • FilesDockWidget.set_select()
      • FilesDockWidget.listwidget_doubleclick()
      • FilesDockWidget.dragEnterEvent()
      • FilesDockWidget.dropEvent()
  • ISAT.widgets.info_dock_widget module
    • InfoDockWidget
      • InfoDockWidget.note_changed()
      • InfoDockWidget.update_widget()
  • ISAT.widgets.mainwindow module
    • QtBoxStyleProgressBar
    • calculate_area()
    • SegAnyThread
      • SegAnyThread.tag
      • SegAnyThread.results_dict
      • SegAnyThread.index
      • SegAnyThread.tag
      • SegAnyThread.sam_encoder()
      • SegAnyThread.run()
    • SegAnyVideoThread
      • SegAnyVideoThread.tag
      • SegAnyVideoThread.start_frame_idx
      • SegAnyVideoThread.max_frame_num_to_track
      • SegAnyVideoThread.tag
      • SegAnyVideoThread.run()
    • InitSegAnyThread
      • InitSegAnyThread.tag
      • InitSegAnyThread.model_path
      • InitSegAnyThread.tag
      • InitSegAnyThread.run()
    • CheckLatestVersionThread
      • CheckLatestVersionThread.tag
      • CheckLatestVersionThread.tag
      • CheckLatestVersionThread.run()
    • MainWindow
      • MainWindow.image_root
      • MainWindow.label_root
      • MainWindow.files_list
      • MainWindow.current_index
      • MainWindow.current_group
      • MainWindow.config_file
      • MainWindow.software_config_file
      • MainWindow.saved
      • MainWindow.can_be_annotated
      • MainWindow.load_finished
      • MainWindow.polygons
      • MainWindow.instance_cmap
      • MainWindow.map_mode
      • MainWindow.current_label
      • MainWindow.use_segment_anything
      • MainWindow.use_segment_anything_video
      • MainWindow.use_remote_sam
      • MainWindow.group_select_mode
      • MainWindow.init_segment_anything()
      • MainWindow.init_sam_finish()
      • MainWindow.sam_encoder_finish()
      • MainWindow.SeganyEnabled()
      • MainWindow.seg_video_start()
      • MainWindow.seg_video_finish()
      • MainWindow.predict_current_image_with_text_prompt()
      • MainWindow.predict_current_image_with_visual_prompt()
      • MainWindow.mask_to_polygon()
      • MainWindow.init_ui()
      • MainWindow.statusbar_change_status()
      • MainWindow.translate()
      • MainWindow.change_language()
      • MainWindow.reload_cfg()
      • MainWindow.load_actions_shortcut()
      • MainWindow.set_saved_state()
      • MainWindow.open_dir()
      • MainWindow.save_dir()
      • MainWindow.save()
      • MainWindow.update_group_display()
      • MainWindow.show_image()
      • MainWindow.prev_image()
      • MainWindow.next_image()
      • MainWindow.jump_to()
      • MainWindow.category_setting()
      • MainWindow.change_bit_map_to_semantic()
      • MainWindow.change_bit_map_to_instance()
      • MainWindow.change_bit_map_to_label()
      • MainWindow.change_bit_map()
      • MainWindow.set_labels_visible()
      • MainWindow.model_manage()
      • MainWindow.remote_sam()
      • MainWindow.change_bfloat16_state()
      • MainWindow.change_use_video_segmentation_state()
      • MainWindow.change_contour_mode()
      • MainWindow.change_contour_method()
      • MainWindow.change_mask_alpha()
      • MainWindow.change_polygon_alpha_hover()
      • MainWindow.change_polygon_alpha_no_hover()
      • MainWindow.change_vertex_size()
      • MainWindow.change_auto_save_state()
      • MainWindow.change_real_time_area_state()
      • MainWindow.change_edge_state()
      • MainWindow.change_approx_polygon_state()
      • MainWindow.change_create_mode_invisible_polygon_state()
      • MainWindow.change_saturation()
      • MainWindow.change_prompt_visiable()
      • MainWindow.converter()
      • MainWindow.video2frames()
      • MainWindow.auto_segment()
      • MainWindow.annos_validator()
      • MainWindow.process_exif()
      • MainWindow.shortcut()
      • MainWindow.about()
      • MainWindow.setting()
      • MainWindow.screen_shot()
      • MainWindow.save_cfg()
      • MainWindow.save_software_cfg()
      • MainWindow.open_docs()
      • MainWindow.exit()
      • MainWindow.closeEvent()
      • MainWindow.latest_version_tip()
      • MainWindow.create_desktop_shortcut()
      • MainWindow.init_connect()
      • MainWindow.reset_action()
  • ISAT.widgets.model_manager_dialog module
    • DownloadThread
      • DownloadThread.urls
      • DownloadThread.name
      • DownloadThread.pause
      • DownloadThread.tag
      • DownloadThread.setNameAndUrl()
      • DownloadThread.run()
      • DownloadThread.move_from_tmp()
    • ModelManagerDialog
      • ModelManagerDialog.init_ui()
      • ModelManagerDialog.update_ui()
      • ModelManagerDialog.download()
      • ModelManagerDialog.download_process()
      • ModelManagerDialog.pause()
      • ModelManagerDialog.delete()
      • ModelManagerDialog.clear_tmp()
      • ModelManagerDialog.open_checkpoint_dir()
      • ModelManagerDialog.load_fine_tuned_model()
      • ModelManagerDialog.use_fine_tune_model()
  • ISAT.widgets.plugin_base module
    • PluginBase
      • PluginBase.init_plugin()
      • PluginBase.enable_plugin()
      • PluginBase.disable_plugin()
      • PluginBase.get_plugin_author()
      • PluginBase.get_plugin_version()
      • PluginBase.get_plugin_description()
      • PluginBase.get_plugin_name()
      • PluginBase.activate_state_changed()
      • PluginBase.before_image_open_event()
      • PluginBase.after_image_open_event()
      • PluginBase.before_annotation_start_event()
      • PluginBase.after_annotation_created_event()
      • PluginBase.after_annotation_changed_event()
      • PluginBase.before_annotations_save_event()
      • PluginBase.after_annotations_saved_event()
      • PluginBase.after_sam_encode_finished_event()
      • PluginBase.on_mouse_move_event()
      • PluginBase.on_mouse_press_event()
      • PluginBase.on_mouse_release_event()
      • PluginBase.on_mouse_pressed_and_mouse_move_event()
      • PluginBase.application_start_event()
      • PluginBase.application_shutdown_event()
  • ISAT.widgets.plugin_manager_dialog module
    • PluginManagerDialog
      • PluginManagerDialog.load_plugins()
      • PluginManagerDialog.update_gui()
      • PluginManagerDialog.trigger_before_image_open()
      • PluginManagerDialog.trigger_after_image_open()
      • PluginManagerDialog.trigger_before_annotation_start()
      • PluginManagerDialog.trigger_after_annotation_created()
      • PluginManagerDialog.trigger_after_annotation_changed()
      • PluginManagerDialog.trigger_before_annotations_save()
      • PluginManagerDialog.trigger_after_annotations_saved()
      • PluginManagerDialog.trigger_after_sam_encode_finished()
      • PluginManagerDialog.trigger_on_mouse_move()
      • PluginManagerDialog.trigger_on_mouse_release()
      • PluginManagerDialog.trigger_on_mouse_press()
      • PluginManagerDialog.trigger_on_mouse_pressed_and_mouse_move()
      • PluginManagerDialog.trigger_application_start()
      • PluginManagerDialog.trigger_application_shutdown()
  • ISAT.widgets.polygon module
    • PromptPoint
    • Vertex
      • Vertex.setColor()
      • Vertex.itemChange()
      • Vertex.hoverEnterEvent()
      • Vertex.hoverLeaveEvent()
    • Polygon
      • Polygon.line_width
      • Polygon.hover_alpha
      • Polygon.nohover_alpha
      • Polygon.points
      • Polygon.vertices
      • Polygon.is_drawing
      • Polygon.category
      • Polygon.group
      • Polygon.iscrowd
      • Polygon.note
      • Polygon.area
      • Polygon.addPoint()
      • Polygon.movePoint()
      • Polygon.removePoint()
      • Polygon.delete()
      • Polygon.moveVertex()
      • Polygon.itemChange()
      • Polygon.hoverEnterEvent()
      • Polygon.hoverLeaveEvent()
      • Polygon.mouseDoubleClickEvent()
      • Polygon.redraw()
      • Polygon.change_color()
      • Polygon.set_drawed()
      • Polygon.calculate_area()
      • Polygon.load_object()
      • Polygon.to_object()
    • LineVertex
      • LineVertex.itemChange()
    • Line
      • Line.addPoint()
      • Line.movePoint()
      • Line.removePoint()
      • Line.delete()
      • Line.redraw()
    • RectVertex
      • RectVertex.itemChange()
    • Rect
      • Rect.addPoint()
      • Rect.movePoint()
      • Rect.removePoint()
      • Rect.delete()
      • Rect.redraw()
  • ISAT.widgets.process_exif_dialog module
    • ProcessExifDialog
      • ProcessExifDialog.open_dir()
      • ProcessExifDialog.start()
  • ISAT.widgets.remote_sam_dialog module
    • RemoteSamDialog
      • RemoteSamDialog.check_connection()
      • RemoteSamDialog.use_remote_sam()
    • IPv4Validator
      • IPv4Validator.validate()
  • ISAT.widgets.right_button_menu module
    • RightButtonMenu
  • ISAT.widgets.setting_dialog module
    • SettingDialog
      • SettingDialog.contour_mode_index_changed()
      • SettingDialog.contour_method_index_changed()
  • ISAT.widgets.shortcut_dialog module
    • ShortcutDialog
      • ShortcutDialog.update_ui()
      • ShortcutDialog.reset_shortcut()
      • ShortcutDialog.shortcut_change_finish()
  • ISAT.widgets.switch_button module
    • SwitchBtn
      • SwitchBtn.checkedChanged
      • SwitchBtn.updateValue()
      • SwitchBtn.mousePressEvent()
      • SwitchBtn.paintEvent()
      • SwitchBtn.drawText()
      • SwitchBtn.drawBg()
      • SwitchBtn.drawSlider()
      • SwitchBtn.setChecked()
  • ISAT.widgets.text_prompt_dock_widget module
    • TextPromptDockWidget
      • TextPromptDockWidget.predict_for_single_category()
      • TextPromptDockWidget.predict_for_multi_categories()
  • ISAT.widgets.video_to_frames_dialog module
    • Video2FramesDialog
      • Video2FramesDialog.open_file()
      • Video2FramesDialog.open_dir()
      • Video2FramesDialog.start()
  • ISAT.widgets.visuall_prompt_dock_widget module
    • VisualPromptDockWidget
  • Module contents

Submodules

ISAT.annotation module

class ISAT.annotation.Object(category, group, segmentation, area, layer, bbox, iscrowd=False, note='')

Bases: object

A class to represent an annotation object.

Parameters:

• category (str) – The category of the object.

• group (int) – The group of the object.

• segmentation (list | tuple) – The vertices of the object.[(x1, y1), (x2, y2), …]

• area (float) – The area of the object.

• layer (int) – The layer of the object.

• bbox (list | tuple) – The bbox of the object. [xmin, ymin, xmax, ymax]

• iscrowd (bool) – The crowd tag of the object.

• note (str) – The note of the object.

class ISAT.annotation.Annotation(image_path, label_path)

Bases: object

A class to represent an annotation containing many objects.

Parameters:

• image_path (str) – The path to the image.

• label_path (str) – The path to the label file.

description

Always ‘ISAT’.

Type:

str

img_folder

The path to the folder where the images are located.

Type:

str

img_name

The name of the image.

Type:

str

label_path

The path to the label file.

Type:

str

note

The note of the image.

Type:

str

height

The height of the image.

Type:

int

width

The width of the image.

Type:

int

depth

The depth of the image.

Type:

int

load_annotation()

Load annotation from self.label_path

save_annotation()

Save annotation to self.label_path

ISAT.configs module

ISAT.configs.load_config(file)

Load config file

Parameters:

file (str) – config file path

Returns:

config dict

Return type:

dict

ISAT.configs.save_config(cfg, file)

Save config file

Parameters:

• cfg (dict) – config dict

• file (str) – config file path

Return type:

None

class ISAT.configs.STATUSMode(value)

Bases: Enum

The status mode

VIEW = 0

CREATE = 1

EDIT = 2

REPAINT = 3

class ISAT.configs.DRAWMode(value)

Bases: Enum

The draw mode.

POLYGON = 0

Manually draw polygon

SEGMENTANYTHING_POINT = 1

Segment anything with point prompt.

SEGMENTANYTHING_BOX = 2

Segment anything with box prompt.

SEGMENTANYTHING_VISUAL = 3

Segment anything with visual prompt.

class ISAT.configs.MAPMode(value)

Bases: Enum

Canvas show map mode

LABEL = 0

SEMANTIC = 1

INSTANCE = 2

class ISAT.configs.CONTOURMode(value)

Bases: Enum

Contour Mode - ways to convert masks to polygons.

SAVE_MAX_ONLY = 0

Only save max contour.

SAVE_EXTERNAL = 1

Only save external contour.

SAVE_ALL = 2

Only save all contour.

ISAT.main module

ISAT.main.main()

Module contents