search

TooTouch / MemSeg

Unofficial re-implementation of MemSeg for Anomaly Detection
MIT License
184 stars 31 forks source link

Can't train if width does not equal height #2

Closed nmddc0211 closed 1 year ago

nmddc0211 commented 1 year ago

Hi TooTouch,

I tried to train your network on my custom images, whose width and height are not equal. I get broadcasting errors if I changed the width and height in the config file. I think you have mixed up some parameters.

TooTouch commented 1 year ago

Hi, @nmddc0211

Thank you for sharing issue. I find the error and modify data/dataset.py to use the custom images, whose width and height are not equal.

Let me know it there are any other issues.

nmddc0211 commented 1 year ago

I think that did it. Thanks a lot!