{"id":23243696,"url":"https://github.com/krik8235/ml-image-segmentation","last_synced_at":"2026-05-09T14:32:36.853Z","repository":{"id":260425552,"uuid":"868081369","full_name":"krik8235/ml-image-segmentation","owner":"krik8235","description":"Build an Image Segmentation Model for Medical Diagnosis","archived":false,"fork":false,"pushed_at":"2024-10-05T15:57:24.000Z","size":27840,"stargazers_count":0,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-02-12T05:45:12.684Z","etag":null,"topics":["cnn-classification","image-classification","ipykernel","jupyter-notebook","machine-learning","machine-learning-algorithms","machine-learning-projects","opencv","opencv-python","pytorch"],"latest_commit_sha":null,"homepage":"https://kuriko.vercel.app/ml-projects/image-segmentation-model","language":"Jupyter Notebook","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/krik8235.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-10-05T12:33:40.000Z","updated_at":"2024-12-13T17:31:04.000Z","dependencies_parsed_at":"2024-10-31T09:28:50.627Z","dependency_job_id":null,"html_url":"https://github.com/krik8235/ml-image-segmentation","commit_stats":null,"previous_names":["krik8235/ml-image-segmentation"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/krik8235%2Fml-image-segmentation","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/krik8235%2Fml-image-segmentation/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/krik8235%2Fml-image-segmentation/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/krik8235%2Fml-image-segmentation/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/krik8235","download_url":"https://codeload.github.com/krik8235/ml-image-segmentation/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247414930,"owners_count":20935316,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["cnn-classification","image-classification","ipykernel","jupyter-notebook","machine-learning","machine-learning-algorithms","machine-learning-projects","opencv","opencv-python","pytorch"],"created_at":"2024-12-19T06:17:43.152Z","updated_at":"2026-05-09T14:32:36.753Z","avatar_url":"https://github.com/krik8235.png","language":"Jupyter Notebook","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Build an Image Segmentation Model for Medical Diagnosis using CNN, PyTorch and OpenCV\n\nIn healthcare and medical science, the fusion of artificial intelligence and deep learning is revolutionizing diagnostics. My project focuses on the precise segmentation of polyps from colonoscopy images—a vital tool for medical practitioners.\n\nIn this project, we aim to develop an image segmentation model using CNN, and deploy it with PyTorch.\n\n\n\n## Data\n\nThe use of the CVC-Clinic database, containing frames from colonoscopy videos. The dataset includes polyp frames and corresponding ground truth images in both PNG and TIFF formats.\n\n\n## Tech stack\n\n- Python 3.12.7\n- **Deep Learning**: PyTorch\n- **Computer Vision**: OpenCV (python-opencv)\n- **Libraries**: Scikit-learn, Pandas, NumPy, Albumentations, YAML (data serialization)\n\n\n## Start in the local env\n\n* Start with installing jupyter environment and dependencies.\n```\n$ pip install ipykernel\n$ pip install -r requirements.txt\n```\n\n* Folder structure\n  - .venv\n  - dataset\n  - ml_deploy\n  - output\n\n\n\n## Takeaways\nAfter 100 epochs of training, \n![Actual image]('https://github.com/krik8235/ml-image-segmentation/blob/main/output/test-images/act.png?raw=true')\n![Segmentation image]('https://github.com/krik8235/ml-image-segmentation/blob/main/output/test-images/seg.png?raw=true')\n\n\n- Polyp Segmentation Insights\n- IOU Metric Understanding\n- Data Augmentation Techniques\n- Practical PyTorch Data Augmentation\n- Medical Computer Vision Applications\n- Building CNN Models\n- Training CNN Models (300 epochs)\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fkrik8235%2Fml-image-segmentation","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fkrik8235%2Fml-image-segmentation","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fkrik8235%2Fml-image-segmentation/lists"}