Dataset-Tools is a desktop application designed to help users browse and manage their image and text datasets, particularly those used with AI art generation tools like Stable Diffusion. Developed using PyQt6, it provides a simple and intuitive graphical interface for browsing images, viewing metadata, and examining associated text prompts. This project is inspired by tools within the AI art community and aims to empower users in improving their dataset curation workflow.
-
Ensure Python is installed on your system.
-
Clone or download the Dataset-Tools repository from GitHub.
-
Install the required dependencies:
pip install PyQt6 pypng
-
Navigate to the directory where
main.pyis located. -
Run the application using Python:
python main.py
The application window has the following main components:
- Current Folder: Displays the path of the currently loaded folder.
- Open Folder: A button to select a folder containing images and text files.
- Image List: Displays a list of images and text files found in the selected folder.
- Image Preview: An area to display a selected image.
- Metadata Box: A text area to display the extracted metadata from the selected image (including Stable Diffusion prompt, settings, etc.).
- Prompt Text: A text label to display the prompt from the selected image.
- Text File Content Area: A text area to display the content of any associated text files.
- Selecting Images: Click on an image or text file in the list to display its preview, metadata, and associated text content.
- Viewing Metadata: Metadata associated with the selected image is displayed on the text area, such as steps, samplers, seeds, and more.
- Viewing Text: The content of any text file associated with the selected image is displayed on the text box.
- Graphical User Interface (GUI): Built with PyQt6 for a modern and cross-platform experience.
- Image Previews: Quickly view images in a dedicated preview area.
- Metadata Extraction: Extract and display relevant metadata from PNG image files, especially those generated from Stable Diffusion.
- Text Viewing: Display the content of text files.
- Clear Layout: A simple and intuitive layout, with list view on the left, and preview on the right.
- Thumbnail Generation: Implement thumbnails for faster browsing.
- JPEG Metadata: Add support for extracting metadata from JPEG files.
- Themes: Introduce customizable themes for appearance.
- Filtering/Sorting: Options to filter and sort files.
- Better User Experience: Test on different operating systems and screen resolutions to optimize user experience.
- Video Tutorials: Create video tutorials to show users how to use the program.
- Text Tutorials: Create detailed tutorials in text and image to show the user how to use the program.
Dataset-Tools is developed by the Duskfall Portal Crew, a diverse system navigating life with DID, ADHD, Autism, and CPTSD. Our motivation stems from leveraging AI to promote inclusivity and mental health awareness. Join us in exploring identity and creative expression through technology.
Thank you for being part of our journey and supporting the Earth and Dusk community.
- ChatGPT 3.5 & 4o: Powering innovative solutions and creative endeavors.
- Support of my peers, and the community at Large.
- Canvas icons created by Freepik - Flaticon
- Inspired by receyuki/stable-diffusion-prompt-reader
Here you can see some screenshots of previous versions of the application.
