Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

51 Commits
 
 
 
 
 
 
 
 

Repository files navigation

🏷️ What's ADE? (Auto Data Embedding) - In Progress


"Train. Detect. Automate."

A PyQt-based desktop app for automatic image labeling and embedding extraction using detection and vision models.

This tool helps you annotate image datasets quickly using pretrained models (e.g., YOLOv8~), modify labels with a GUI, and extract embeddings for downstream tasks.




Main View
Screenshot

Auto Embedding. . .
Screenshot

Train hyperparameter setting
Screenshot




📦 Features (planned)

✅ Automatic object detection & YOLO format label generation

✅ Visual label editing using PyQt GUI

✅ Embedding extraction (e.g., with CLIP or ResNet)

✅ Save embeddings as .npy, .json

✅ Support for .jpg, .png, .txt YOLO datasets




🔧 Setup Instructions

1. Download Release version

Release Date Version
2025-06-12 - Recent v1.0.2
2025-06-10 v1.0.1
2025-06-05 v1.0.0

2. Create and Activate a Virtual Environment

macOS/Linux

python3 -m venv venv
source venv/bin/activate

Windows

python -m venv venv
venv\Scripts\activate

3. Install Required Packages

pip install -r requirements.txt

Example requirements.txt

PyQt5==5.15.11
opencv-python==4.11.0.86
ultralytics==8.3.145
numpy==1.24.4
pillow==10.4.0
matplotlib==3.7.5

4. Install CUDA pytorch

pip install torch==2.4.1+cu121 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121




▶️ How to Build

pyinstaller main.py --onedir --noconsole --clean --icon=icons/icon.ico --add-binary "yolo.exe;."




📁 Project Structure

pyqt_app/
├── build_guide.txt               # Build guide
├── main.py                       # App entry point
├── data_integrator.py
├── training_settings_dialog.py
├── icons/                        # PyQt GUI icons
├── model/                        # YOLO, CLIP, etc. wrappers
├── dist/
│   └── main.exe                  # .exe file
├── dataset/
│   ├── images/
│   └── labels/
├── requirements.txt              # Requirements package settings
└── README.md




🧠 Model Support

  • YOLOv11 via Ultralytics
  • CLIP (planned)
  • ResNet (planned)
  • ONNX custom model support (planned)




🙌 Contributing

Issues, feedback, and pull requests are welcome! Open a GitHub issue or fork this repo to contribute.



About

PyQt 기반의 데스크탑 애플리케이션으로, YOLO 데이터셋 자동 라벨링과 임베딩 추출을 지원합니다. 사전 학습된 객체 검출 모델을 활용해 빠르고 손쉽게 이미지에 라벨을 생성하고, GUI를 통해 라벨을 직관적으로 수정할 수 있으며, 다양한 컴퓨터 비전 작업에 활용할 수 있는 임베딩까지 추출할 수 있습니다.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages