Devanik21 / ISRO_Mining_Site_FINAL_APP

MIT License
9 stars 37 forks source link

U-Net | Predicting Lunar-Craters Using Deep-Learning #93

Open Varunshiyam opened 1 day ago

Varunshiyam commented 1 day ago

This project developed a U-Net model for image segmentation, specifically for identifying lunar craters in images. The model was trained on a dataset of lunar images and corresponding masks, with data augmentation techniques used to improve model robustness. The model was compiled with the Adam optimizer and a dice coefficient loss function, and evaluated using metrics such as IOU and dice coefficient. The trained model was then tested on a separate dataset of lunar images, and also applied to images of lunar boulders to assess its generalization capability.

Kindly, Assign me Under :

Avishek8136 commented 1 day ago

I can do this Assign this to me