search

wentaozhu / deep-mil-for-whole-mammogram-classification

Zhu, Wentao, Qi Lou, Yeeleng Scott Vang, and Xiaohui Xie. "Deep Multi-instance Networks with Sparse Label Assignment for Whole Mammogram Classification." MICCAI 2017.
MIT License
113 stars 36 forks source link

running error #4

Closed yishanyang closed 7 years ago

yishanyang commented 7 years ago

Hi, wentao Here is an error when I run the inbreast.py :

Traceback (most recent call last): File "inbreast.py", line 292, in traindata, trainlabel, testdata, testlabel = loaddata(0, 5) File "inbreast.py", line 200, in loaddata ims = loadim(mydictkey[trainindex[i]]+'.pickle', aug=aug) TypeError: loadim() got an unexpected keyword argument 'aug'

It will be grateful if you can help me to solve the problem.

Danyang Li

wentaozhu commented 7 years ago

Hi Danyang,

You can try to comment the aug parameter.

Thanks, Wentao

yishanyang commented 7 years ago

Hi wentao

I have solved it. Thanks very much!

Danyang Li