labelaug is a simple python GUI app that is written in python and uses QT for graphical interface. It allows user to easily augment images and their corresponding labels in YOLO format. It offers multiple basic augmentation techniques which are most commonly used for data augmentation. Users can also see the bounding boxes on their images as well as export the image as a .jpg file. It takes care of augmentation of multiple images and labels in a given directory without any sweat!
Python 3.0 or above Clone the repository
git clone https://github.com/Munimul/LabelAug.git
Change directory (cd) to the repository where it is cloned
cd LabelAug
Create a virtual Environment and acitvate it before installing dependencies(Optional)
Install Dependencies
pip install -r requirements.txt
Run the labelaug.py file
python labelaug.py
Follow the steps above and run the labelaug.py file
- Select the directory using 'Label Open Directory' button where your label or yolo.txt files are located
- If the images are located in different folder, select that folder using 'Image Open Directory' button or select the same directory
- Select a folder where you want to save the augmented images and labels
- All the image can be displayed with their corresponding labels. Navigate by 'Next' and 'Previous' button
- Current image displayed on the GUI can be saved as a .jpg file.
- Select atleast one or multiple augmentation method
- Push 'Go' button
- For every augmentation, a subfolder will be created in your selected save directory. Images and label files will be located there.
LabelAug can only work with valid Yolo label format in the .txt file which is : class, x_center, y_center, width , height The class should be an integer value and rest of the four in normalized(0-1) float value.
Sample user interface displaying all directory information and contained images and labels information
