module 'mrcnn.model' has no attribute 'data_generator'
Closed this issue · 2 comments
neo1zh commented
I'm newly trying your mrcnn model with tf2.7. The demo ran successfully but when i was trying to train balloon, this error occured.
Here's my trace back:
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
d:\proj\Mask-RCNN-TF2.7.0-keras2.7.0-main\samples\balloon\inspect_balloon_data.ipynb Cell 24 in <cell line: 3>()
[1](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=0) # Create data generator
[2](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=1) random_rois = 2000
----> [3](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=2) g = modellib.data_generator(
[4](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=3) dataset, config, shuffle=True, random_rois=random_rois,
[5](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=4) batch_size=4,
[6](vscode-notebook-cell:/d%3A/proj/Mask-RCNN-TF2.7.0-keras2.7.0-main/samples/balloon/inspect_balloon_data.ipynb#X31sZmlsZQ%3D%3D?line=5) detection_targets=True)
AttributeError: module 'mrcnn.model' has no attribute 'data_generator'
neo1zh commented
ohhhh sorry this is my first issue on this community, forgive my ignorance of greeting.
Seems same as this issue matterport/Mask_RCNN#2112, but not solved yet.
Thanks,
Zylone
Kamlesh364 commented
Use balloon.py instead.