Mask predictions

[Abdullah955]

I've installed and tested the model on Box prediction and it works fine, when i tried to switch to masks referring to this page https://github.com/MendelXu/SoftTeacher/blob/main/ssod/models/mask_soft_teacher.py

where can i switch from the soft_teacher.py to mask_soft_teacher.py ?

[MendelXu]

You can refer to https://github.com/microsoft/SoftTeacher/issues/146 to modify your config (change semi_wrapper.type to MaskSoftTeacher)and add from .mask_soft_teacher import MaskSoftTeacher in ssod/models/__init__.py

[Abdullah955]

I'm facing ArrtibuteError: 'StandardRoIHead' object has no attribute 'mask_head' do you think it's a version problem ?

[MendelXu]

No. Please refer to #146.

[Abdullah955]

I've already edited the base file still have same problems as this issue

https://github.com/microsoft/SoftTeacher/issues/146 TypeError: 'NoneType' object is not iterable

and i've added the same roi_head configurations you mentioned.

[MendelXu]

OK. Thanks for your reply. I will post a correct config file here tonight.

[MendelXu]

Putting this file(https://gist.github.com/MendelXu/143b6b46ac35796785aae9d4087393de) in configs/soft_teacher/ should work.

[Abdullah955]

it worked! Thank you for great support

[alaa-shubbak]

Hello , I did the same things . and change the config as well as the following command

You can refer to #146 to modify your config (change semi_wrapper.type to MaskSoftTeacher)and add from .mask_soft_teacher import MaskSoftTeacher in ssod/models/__init__.py

but unfortunatily , I still get an error , I have this error : KeyError: 'MaskSoftTeacher is not in the models registry'

it is an error about register stuff , how can i solve it?