search

Kohulan / DECIMER-Image-Segmentation

Chemical structure detection and segmentation tool for Journal articles.
https://decimer.ai
MIT License
81 stars 30 forks source link

Issues with mrcnn parallel model #109

Closed cmc227 closed 2 weeks ago

cmc227 commented 2 months ago

Hi,

When running across multiple gpus, I get ModuleNotFoundError from mrcnn.parallel_model import ParallelModel in model.py.

I add the parallel model from https://github.com/matterport/Mask_RCNN but then get the error 'ParallelModel' object has no attribute '_base_model_initialized'

Would anyone be able to help with this?

Thanks

Kohulan commented 2 months ago

Hi @cmc227 ,

I will try to look into this. Since it is Mask RCNN issue I think it is due to TF 2.0. The original Mask RCNN was written using TF 1.0.