VAME icon indicating copy to clipboard operation
VAME copied to clipboard

Added option to evaluate all snapshots, fixed bug when prediction_dec…

Open alexcwsmith opened this issue 4 years ago • 0 comments

I added a use_snapshots boolean parameter to vame.evaluate_model, when set to True it will evaluate all snapshots (similar to DeepLabCut) which allows the user to get a better look at how training went. It would benefit from more documentation that if they want to use a snapshot rather than what is considered "best model" you would just move that .pkl file into the 'best_model' folder.

I also fixed a typo at the end that still directs users to next use vame.behavior_quantification, which is from the old version, I believe now the next step is to use vame.pose_segmentation.

Lastly, there was a bug in running evaluate_model when prediction_decoder was off (it would return a blank reconstruction image). I fixed that bug.

alexcwsmith avatar Oct 12 '21 17:10 alexcwsmith