search

teamtomo / fidder

Detect and erase gold fiducials in cryo-EM images
https://teamtomo.org/fidder/
Other
11 stars 4 forks source link

AttributeError: module 'torchvision.transforms.functional' has no attribute 'get_dimensions' #13

Closed JulienMaufront closed 1 year ago

JulienMaufront commented 1 year ago

Hello,

When I try to use "fidder predict" command, I get an error message : "AttributeError: module 'torchvision.transforms.functional' has no attribute 'get_dimensions'" Do you know how I can set this up ?

Thank you, Cheers

Julien Maufront

alisterburt commented 1 year ago

Hey Julien,

Weird - I thought maybe there was a torchvision API change but the function is still there in their source

https://github.com/pytorch/vision/blob/a192c95e77a4a4de3a8aeee45130ddc4d2773a83/torchvision/transforms/functional.py#L64

I'll take a quick look and push a fix this afternoon for you!

JulienMaufront commented 1 year ago

Oh that's wonderful, thank you !

JulienMaufront commented 1 year ago

Hello Alister,

I'm so sorry, actually I just had to upgrade my pytorch version and now it works like a charm !
I hope I haven't wasted too much of your time making you look for this issue.

Thanks again !

alisterburt commented 1 year ago

Hey Julien! Thanks for reporting back in - I did some quick tests and couldn't see anything wrong but wanted to check a bit more before telling you it was something your end, glad it's working for you!

Would be curious to hear about performance too, feel free to drop me an email with any thoughts!