Skip to content

FIX #143 - issues with labelme2coco Dataset in MaskRCNN custom train#144

Open
mabu-dev wants to merge 1 commit intoayoolaolafenwa:masterfrom
mabu-dev:fix_MaskRCNN_labelme2coco
Open

FIX #143 - issues with labelme2coco Dataset in MaskRCNN custom train#144
mabu-dev wants to merge 1 commit intoayoolaolafenwa:masterfrom
mabu-dev:fix_MaskRCNN_labelme2coco

Conversation

@mabu-dev
Copy link

This fixes #143

Fixes include:

  1. Make load_dataset compatible with labelme2coco.convert()
  2. Ensures that labelme2coco.convert() respects the 0 class ID reservation and avoid throwing:
if class_id < 1:
print(
'Error: Class id for "{}" cannot be less than one. (0 is reserved for the background)'.format(class_name)
)
  1. Updates docs for custom train example

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Custom MaskRCNN error

1 participant