iteal / wormpose

WormPose: Image synthesis and convolutional networks for pose estimation in C. elegans
https://iteal.github.io/wormpose/
BSD 3-Clause "New" or "Revised" License
46 stars 16 forks source link

Dataset loader not found in google colab tutorial notebook #24

Closed meldefon closed 1 year ago

meldefon commented 1 year ago

Hello! Thanks for this repository. It looks very cool!

I was trying to run the tutorial notebook on google colab. Didn't change anything, just ran the cells as is, and got as far as the "Set inputs" step before I got the error

NotImplementedError: Dataset loader: 'sample_data' not found in the package entry point

I tried changing the dataloader to "tierpsy". No luck there either. Do you have any idea what might be going on here?

Thank you!

iteal commented 1 year ago

Hi @meldefon ! Usually this error in Colab meant you need to restart the Runtime after installing the package wormpose with pip. Can you confirm you installed and then restarted the Runtime in google colab? Thanks!

meldefon commented 1 year ago

Thank you so much for getting back to me! I didn't realize I had to restart the runtime. I'm able to run those cells now - problem solved!