Skip to content

YeLyuUT/UAVidToolKit

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.

UAVidToolKit

UAVidToolKit provides basic tools for easier usage of the UAVid dataset. Including label conversion, label visualization, performance evaluation and image label path pair txtfile preparation.

Install

Download the toolkit into the dataset folder as follows,

cd <UAVid dataset folder>
git clone https://github.com/YeLyuUT/UAVidToolKit.git
cd UAVidToolKit
python setup.py build_ext --inplace
cd ..

Rename training, validation and testing subfolders into 'train', 'valid' and 'test'. Or create symlink with cmd,

ln -s <train dir> train
ln -s <valid dir> valid
ln -s <test dir> test

The data structure should be like:

\UAVidDataset
    \train
        \seq
        ...
    \valid
        \seq
        ...
    \test
        \seq
        ...
    \UAVidToolKit

Usage

In the UAVid dataset folder, apply commands as follows:

  • Label image conversion from 3 channel RGB color image to 1 channel label index image.
python UAVidToolKit/prepareTrainIdFiles.py -s <src folder> -t <dst folder>

e.g. python UAVidToolKit/prepareTrainIdFiles.py -s valid/ -t tooltest/

  • Label image conversion from 1 channel label index image to 3 channel RGB color image.
python UAVidToolKit/convertTrainIdFiles2Color.py -s <src folder> -t <dst folder> -f <sub folder name>

e.g. python UAVidToolKit/convertTrainIdFiles2Color.py -s tooltest/ -t tooltest/ -f 'color'

  • Blend image and label files.
python UAVidToolKit/blendImageAndLabel.py -i <image folder> -l <label folder> -o <output folder> -id <image subfolder name> -ld <label subfolder name> -od <output subfolder name>

e.g. python UAVidToolKit/blendImageAndLabel.py -i valid/ -l tooltest/ -o tooltest/ -id Images -ld color -od blend

  • Performance evaluation.
python UAVidToolKit/evaluate.py -gt <ground truth folder> -p <prediction folder> -v

If add '-v', visualize mIoU and confusion matrix results with figures.

e.g. python UAVidToolKit/evaluate.py -gt valid -p pred_valid -v

  • Write image label paths pair into txt.
python UAVidToolKit/writeImageLabelPathPairsToTxtFile.py -d <dataset folder> -t -v

If add '-t', add training set to txt.

If add '-v', add valid set to txt.

e.g. python UAVidToolKit/writeImageLabelPathPairsToTxtFile.py -d ./ -t -v

A Message

If you have any question or new feature suggestion, please create an issue to let me know.

About

UAVidToolKit

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published