search

fishial / fish-identification

Fish Detection (Segmentation) & Classification models and training scripts
https://www.fishial.ai
MIT License
41 stars 6 forks source link

Torch Script Model #3

Open Hasnain1997-ai opened 9 months ago

Hasnain1997-ai commented 9 months ago

Hi! In your TorchScript conversion notebook, you took a model named "model_21_08_2023_test.pt" and converted it to TorchScript. However, I am unable to find this model or any other segmentation .pt model in your repository. Could you please guide me on where to find it or what steps to take?

Hasnain1997-ai commented 9 months ago

Also where are these models model_path = r'/home/fishial/Fishial/output/segmentation_export_torchscript/model_segm_21_08_2023.ts'

AndrzejCodahead commented 8 months ago

Hi @Hasnain1997-ai , the pytorch file of segmentation model you could download under this link , but if you loocking for already converted torchscript file you may download from here.

Hasnain1997-ai commented 8 months ago

@AndrzejCodahead Thank you so much for the replay I will try the .ts converted model you share