search

georgeretsi / mushroom-pose

3D pose estimation of mushrooms from pointcloud - trained on synthetic data
3 stars 0 forks source link

where is fnl_model.pt? #1

Closed heidapao2b closed 3 weeks ago

heidapao2b commented 2 months ago

Thank you for your working! I want to know where is fnl_model.pt .

georgeretsi commented 2 months ago

Hi there! Thanks for your interest in our work. The "fnl_model.pt" is not the correct name, since I have renamed this in "trained_model.pt" which exists in the repo. I've updated the code. This is to start training from an already well-performing model, but you can also try removing the load_state_dict commands (this may take several epochs).

heidapao2b commented 2 months ago

OK! Thank you very much！