modifications for custom grayscale dataset (input format .hdf5) - Githubissues

allanzelener / YAD2K

YAD2K: Yet Another Darknet 2 Keras

Other

2.71k stars 877 forks source link

modifications for custom grayscale dataset (input format .hdf5) #140

Open divya-nk opened 6 years ago

divya-nk commented 6 years ago

myds_retrain.py:

works with input format .hdf5 for a custom dataset
modified code to work for HD grayscale image
modified to work for more than one bounding box in an image.
train-validation split implemented in this code

voc_conversion_scripts/convert_to_hdf5.py

converts all images in the custom dataset into a single .hdf5 format file
modified bounding box units to host coordiates of HD images