Repository navigation
about tf_records_generation.py for sentinel-hub example #133
Description
Hello I am looking at the Preparing Data with Sentinel Hub example, since that label-maker also support to prepare the mask annotation of buildings as shown in this examples, do you have some suggestion how to conver the images and lables into tensorflow record for fine-tuning the pre-trained model ?
I have seen some similar python script such as models/research/object_detection/dataset_tools/create_pet_tf_record.py, while to modify this based on label-maker results might be missing :)
Activity
drewbo commentedon Mar 13, 2019
@bobleegogogo have you tried running the
tf_records_generation.pyscript with the Sentinel Hub data? I think the major change you would need to make from the Sentinel Hub example is to usml_typeasobject_detectionbut otherwise this should be similar to the other building detection example.bobleegogogo commentedon Mar 13, 2019
@drewbo Thanks for your reply. Actually, I was trying to use label-maker to prepare training samples for Mask-RCNN, this is why I need to use ml_type as segmentation. Then the tf_records_generation.py cannot meet the demand, since the training samples should contain mask and bounding boxes, not only the bounding boxes. I found some similar script here: create_mask_rcnn_tf_record.py