Build error due to display problems in travis.
End to end software to capture new objects using RGBD camera and augment them to get a annotated dataset to train deep nets
- Pipeline to artificial generate annotated data for training deep learning models.
- Pipeline includes starting from capturing images using provided camera (Realsense), generate semantic labels of the captured image and then generate the artificial images.
- GUI required for the end user from capturing to labelling and generating new data.
- Ubuntu 16.04 (Testing for Ubuntu 18.04)
- Intel Realsense Camera
- Processer intel i5 or higher
- python 3.5
- Number of classes captured should be more than or equal to 2.
- Only one object in the scene to segment, not supported for multiple objects.
Linux:
git clone https://github.com/santoshreddy254/easy_augment.git
cd easy_augment
./setup.sh
pip3 install easy-augment --user
cd realsense_augmentor
python3 src/main.py
- Start page will be as below and select the path to save the captured images.
- On selecting capture next window will look like
- On selectin Have Annotations Steps after selecting capture option and selecting save path
- Next window will have image and mask of corresponding object. Capture as many as images per classe.
- Click add to add new class label.
- Click save to save the current displyed image and semantic label.
- Click finish once done with capturing all the images.
- Folder name \textbf{captured_data} in selected save path will have images, labels and labels.txt
- Input parameters to generate artificial images need to be filled next window.
- Click OK once setting up the parameters.
- Folder name \textbf{augmented} in selected save path will have artificial images.
Steps after selecting Have annotations
-
Select the location of images and corresponding annotations
-
Select save path
-
Select labels.txt file path
-
Follow the steps from Step 9
- 1.0.2
- Fixed bugs with augmeting data with labelme data.
- 1.0.1
- Released on pypi.
- 1.0.0
- First release for crowd testing.
- Santosh Muthireddy – https://github.com/santoshreddy254
- Naresh Kumar Gurulingan - https://github.com/NareshGuru77
- Deepan Chakravarthi Padmanabhan - https://github.com/DeepanChakravarthiPadmanabhan
- M.Sc Deebul Nair - https://github.com/deebuls
Distributed under the MPL license. See LICENSE
for more information.