# few-shot-label-until

**Repository Path**: kkangcode/few-shot-label-until

## Basic Information

- **Project Name**: few-shot-label-until
- **Description**: 一个利用小样本分割的自动标注工具
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 0
- **Created**: 2024-07-31
- **Last Updated**: 2024-10-15

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 小样本自动标注工具

基于pyqt5、pytorch 实现的一个自动标注程序
![detail](./figure/detail.gif)

## 用法

- 创建目录
  
  ```bash
  mkdir -p exp/pascal/BAM/split0/resnet50
  mkdir initmodel
  mkdir -p initmodel/PSPNet
  ```
- 参考  [BAM](https://github.com/chunbolang/BAM) ，下载预训练权重从 [here](https://mailnwpueducn-my.sharepoint.com/:u:/g/personal/langchunbo_mail_nwpu_edu_cn/EflpnBbWaftEum485cNq8v8BdSHiKvXLaX-dBBsbtdnCjg?e=WLcfhd) 之后将其放入 `/initmodel` 文件夹
- 参考  [BAM](https://github.com/chunbolang/BAM) ，下载预训练权重从  [OneDrive](https://mailnwpueducn-my.sharepoint.com/:u:/g/personal/langchunbo_mail_nwpu_edu_cn/ETERT3xe5ndEpDhStts7JmcBlYDY_2G0hPVJUBtLLG-njg?e=MLzVIL) 之后将其放入 `initmodel/PSPNet`.
- 将预训练权重放入`exp/pascal/BAM/split0/resnet50`,训练权重可从 [BAM](https://github.com/chunbolang/BAM) 下载
- 运行
  
  ```bash
  python main.py
  ```

## 主要功能

![show](./figure/show.png)

### 上传图像

支持支持图像、支持标签、查询图像的上传功能

### 选取目标

支持在图像框中选取想分割的目标

### 生成选取标签

通过支持图片及支持图像的标签，在结果生成框中生成查询图像的标签

## 结果展示

![res](./figure/res.png)

## 小样本分割模型使用

本程序目前使用的小样本分割模型为 BAM: https://github.com/chunbolang/BAM   可根据需要自行替换。