Skip to content

Latest commit

 

History

History
60 lines (51 loc) · 2.84 KB

README.md

File metadata and controls

60 lines (51 loc) · 2.84 KB
Raw

vision_dev

Scripts

  • split_data.sh

    This script splits a directory full of images into multiple numbered directories with 100 images each

    • Usage

      ./split_data.sh <source directory>
      • source directory: [string] The directory storing the original set of image

Sloth

Sloth is an image tagging tool.

Configuration

Provided, is the sloth configuration file, robosub_config.py, to specialize sloth for Robosub use. Included, are the required annotation conversions for any frameworks used by robosub.

  • Annotation types

    • Darknet

      Darknet has its own specific format which requires multiple files, one per image. For this, we have created a special *.darknet format to bridge the gap between sloth's one file system to darknet's multiple file system. The *.darknet format is as follows

      ---
<class name> : <class number>
class_name_1 : 0
class_name_2 : 1
      :
      :
---
>> "filename_1.jpg"
<class number> <x_position_ratio> <y_position_ratio> <width_ratio> <height_ratio> //Darknet format line
<<
>> "filename_2.png"
<<
    :
    :
      • class name: [string] The name of the class within sloth for the tag
      • class number: [integer] The number assotiated with the class name used within darknet
      • x_position_ratio: [0.0...1.0] The ratio between the x position of the center of the rectangle and the image width
      • y_position_ratio: [0.0...1.0] The ratio between the y position of the center of the rectangle and the image height
      • width_ratio: [0.0...1.0] The ratio between the width of the rectangle and the image width
      • height_ratio: [0.0...1.0] The ratio between the height of the rectangle and the image height

      Utilize the sloth_to_darknet.py script to generate the darknet files from the above annotation format.

Scripts

  • sloth_to_darknet.py

    Converts from the *.darknet format specified above to multiple files within an output directory and a training list used with Darknet

    • Usage

      ./sloth_to_darknet.py -f <filename> -o <output directory> [-t <training list filename>]
      • filename: [string] The filepath to the *.darknet file from which to extract information.
      • output directory: [string] The directory into which the darknet compatible annotation files are output.
      • (optional) training list filename: [string] The filename of the output training list

  • sloth_to_tfrecord.py

    Converts json format files into tenserflow format.

    • Usage

      ./sloth_to_tfrecord.py <config_file> <output directory>
      • filename: [string] The filepath to the *.json file from which to extract information.
      • output directory: [string] The directory into which the darknet compatible annotation files are output.