search

jxwleong / coral-sleuth

Coral-Sleuth: A personal project employing deep learning techniques to sleuth out different types of corals from images. Dive in to explore the underwater world of ML!
2 stars 1 forks source link

[ask] model h5 #23

Closed isnuryusuf closed 9 months ago

isnuryusuf commented 9 months ago

Hi

can you share file or inform me to generate:

coral_reef_classifier_efficientnetv2_full_epoch_200_1_batchsize_32_combined_annotations_about_40k_png_only_remapped_majority_class_with_3k_to_4k_sample_reduced_1k_scale_0.2.h5

thank

jxwleong commented 9 months ago

Hi @isnuryusuf,

Did you manage to download the full datasets?

You can train the model to generate this file (coral_reef_classifier_efficientnetv2_full_epoch_200_1_batchsize_32_combined_annotations_about_40k_png_only_remapped_majority_class_with_3k_to_4k_sample_reduced_1k_scale_0.2.h5)

or do you mind to share me your email? I can give you access to this file on my Google Drive.

Regards, Jason

isnuryusuf commented 9 months ago

thnks @jxwleong

jxwleong commented 9 months ago

I have shared the file with your Gmail. Have to look and let me know if you need any other information.

isnuryusuf commented 9 months ago

thnks @jxwleong

but i think iam not received your file

jxwleong commented 9 months ago

My bad!

Looks like my organization's Google Drive is preventing it to shared with others outside of the organization and I can't upload it to my personal drive or email it to you directly due to its size.

Would you want to try to train the model and generate the file on your own? I can share the steps later if you're interested.

isnuryusuf commented 9 months ago

Yes sure, thnks

isnuryusuf commented 9 months ago

terima kasih

jxwleong commented 9 months ago

Alright.

Here are the steps (assuming the datasets are downloaded and placed in /data/images):

  1. Change the following variables to the desired settings. Such as combined_annotations_about_40k_png_only_remapped_majority_class_with_3k_to_4k_sample_reduced_1k.csv and 0.2. https://github.com/jxwleong/coral-sleuth/blob/c62dddee3e23e34aa13838a9c52dbdbf66dca0c9/src/model_training.py#L167 https://github.com/jxwleong/coral-sleuth/blob/c62dddee3e23e34aa13838a9c52dbdbf66dca0c9/src/model_training.py#L191

  2. Execute the /src/model_training.py.

I'm running the model training on a dedicated server.