diff --git a/README-CN.md b/README-CN.md index c31940f..4985476 100644 --- a/README-CN.md +++ b/README-CN.md @@ -1,17 +1,19 @@ # ISAT with segment anything # ISAT 图像分割标注工具(集成segment anything) -集成[segment anything](),实现图片分割快速标注。 +集成[segment anything](https://github.com/facebookresearch/segment-anything),实现图片分割快速标注。 ## 特点 -1. segment anything简化标注过程 +1. 集成segment anything,快速进行图像分割标注 - 通过鼠标左右键提示感兴趣区域,调用segment anything自动计算分割掩码。不必再手动进行目标轮廓选取。 + - 通过鼠标左右键提示感兴趣区域,调用segment anything自动计算分割掩码。不必再手动进行目标轮廓选取。 + - 自动生成的掩码转换为多边形,进行手动调整。 -2. 便捷地修改标注结果 - - segment anything只是协助生成目标掩码,最终标注以多边形呈现,通过拖拽多边形顶点,可快速修改标注区域。 +2. 手动绘制多边形进行精细标注 + - 保留了ISAT手动绘制多边形进行标注的功能,可满足Segment anything无法分割目标的标注。 + - 手动标注较自动标注更加精确,但工作量也更大。 + ## 安装 ### 1. 源码运行 ```shell @@ -55,11 +57,13 @@ python main.py ## 标注操作 -1. 通过鼠标左键(或右键)提示感兴趣区域(或不感兴趣区域),自动形成目标分割掩码 -2. 可通过多次左右键提示,提升掩码质量 -3. E键结束标注,选择类别,得到多边形标注区域 -4. 拖拽多边形顶点,精细化调整标注 +1. 通过鼠标左键(或右键)提示感兴趣区域(或不感兴趣区域),自动形成目标分割掩码。 +2. 可通过多次左右键提示,提升掩码质量。 +3. E键结束标注,选择类别,得到多边形标注区域。 +4. 拖拽多边形顶点,精细化调整标注。 ## 注意事项 1. 自动分割效果受segment anything模型分割效果限制,如需更为精确的分割效果,可通过手动绘制多边形实现。 2. 如只需要使用手动绘制多边形标注,推荐使用[ISAT](https://github.com/yatengLG/ISAT)。 +3. 如果没有GPU,不建议使用ISAT_with_segment_anything,载入图片花费时间较长。 +4. 如果GPU显存较小,建议使用sam_vit_b_01ec64模型。 diff --git a/widgets/mainwindow.py b/widgets/mainwindow.py index ccb8fef..535b446 100644 --- a/widgets/mainwindow.py +++ b/widgets/mainwindow.py @@ -66,7 +66,6 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow): self.statusbar.showMessage('Find the checkpoint named {}.'.format('sam_vit_b_01ec64.pth')) self.segany = SegAny('./segment_any/sam_vit_b_01ec64.pth') self.use_segment_anything = True - self.statusbar.showMessage() else: QtWidgets.QMessageBox.warning(self, 'Warning', 'The checkpoint of [Segment anything] not existed. If you want use quick annotate, please download from {}'.format('https://github.com/facebookresearch/segment-anything#model-checkpoints')) self.use_segment_anything = False