adursun / wsddn.pytorch

Implementation of Weakly Supervised Deep Detection Networks using the latest version of PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TypeError: __call__() got an unexpected keyword argument 'image'

Scienthusiasts opened this issue · comments

Hi!I'm trying to run train.py to train the network but I'm having this problem:
TypeError: call() got an unexpected keyword argument 'image'
and It seems that the problem is in this part of utils.py:
image
how can I solve the problem,I'm not sure if it's the code or me, thank you!

Hi,

Please check the version of albumentations package in your local. This code expects albumentations==0.4.3 as in the requirements.txt.

thanks for your reply!but I still got the problem below:

image

Please use the exact versions in https://github.com/adursun/wsddn.pytorch/blob/master/requirements.txt, the new package methods/parameters might cause problems.

Oooh,I got, I have chaged the code ,I reverted the code back to the original and it worked:
image
thanks a lot!

Thank you for your patience!, but I still have a question, in the original paper the author used the SPP module, but it seem that your version doesn‘t use it?