{"id":21345359,"url":"https://github.com/aribornstein/keras-3d-image-classification","last_synced_at":"2026-05-11T09:26:37.749Z","repository":{"id":66190409,"uuid":"356517806","full_name":"aribornstein/Keras-3D-Image-Classification","owner":"aribornstein","description":"Based on the Keras 3D Image Classification from CT Scans by Hasib Zunair from the Keras IO examples repo","archived":false,"fork":false,"pushed_at":"2021-04-14T10:32:26.000Z","size":37,"stargazers_count":3,"open_issues_count":0,"forks_count":1,"subscribers_count":4,"default_branch":"main","last_synced_at":"2025-03-09T19:11:50.277Z","etag":null,"topics":["3dimageclassification","gridai","keras","tensorboard","tensorflow"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/aribornstein.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","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":"2021-04-10T08:19:09.000Z","updated_at":"2023-12-28T09:56:19.000Z","dependencies_parsed_at":null,"dependency_job_id":"32dd38dc-bd4f-43ed-b562-5299ac9b156b","html_url":"https://github.com/aribornstein/Keras-3D-Image-Classification","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aribornstein%2FKeras-3D-Image-Classification","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aribornstein%2FKeras-3D-Image-Classification/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aribornstein%2FKeras-3D-Image-Classification/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aribornstein%2FKeras-3D-Image-Classification/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/aribornstein","download_url":"https://codeload.github.com/aribornstein/Keras-3D-Image-Classification/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":243822313,"owners_count":20353496,"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":["3dimageclassification","gridai","keras","tensorboard","tensorflow"],"created_at":"2024-11-22T01:31:30.238Z","updated_at":"2026-05-11T09:26:37.718Z","avatar_url":"https://github.com/aribornstein.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Keras 3D Image Classification with TensorBoard Logging \n\n\n![3D Brain Scan Segments](https://keras.io/img/examples/vision/3D_image_classification/3D_image_classification_19_0.png)\n\n## Run with Grid\n\n[![Grid](https://img.shields.io/badge/rid_AI-run-78FF96.svg?labelColor=black\u0026logo=data:image/svg%2bxml;base64,PHN2ZyB3aWR0aD0iNDgiIGhlaWdodD0iNDgiIGZpbGw9Im5vbmUiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTEgMTR2MjBhMTQgMTQgMCAwMDE0IDE0aDlWMzYuOEgxMi42VjExaDIyLjV2N2gxMS4yVjE0QTE0IDE0IDAgMDAzMi40IDBIMTVBMTQgMTQgMCAwMDEgMTR6IiBmaWxsPSIjZmZmIi8+PHBhdGggZD0iTTM1LjIgNDhoMTEuMlYyNS41SDIzLjl2MTEuM2gxMS4zVjQ4eiIgZmlsbD0iI2ZmZiIvPjwvc3ZnPg==)](https://platform.grid.ai/#/runs?script=https://github.com/aribornstein/Keras3DImageClassifcation/blob/800e59c500af6a80c692d8a46b43c94505718adf/train.py\u0026cloud=grid\u0026instance=g4dn.xlarge\u0026accelerators=1\u0026disk_size=200\u0026framework=tensorflow\u0026script_args=grid%20train%20train.py%20--g_gpus%201%20--g_instance_type%20g4dn.xlarge%20--g_framework%20tensorflow%20--max_epochs%205)\n\nThis demo is a slighly modifed version of the [Keras 3D Image Classification from CT Scans\n](https://keras.io/examples/vision/3D_image_classification/) by [Hasib Zunair](https://hasibzunair.github.io/) from the [Keras IO examples repo](https://github.com/keras-team/keras-io/tree/master/examples). I have added tensorboard logging and argparse to enable more training customization and hyperparameter tuning. \n\nThe example shows the steps needed to build a 3D convolutional neural network (CNN) to predict the presence of viral pneumonia in computer tomography (CT) scans. 2D CNNs are commonly used to process RGB images (3 channels). A 3D CNN is simply the 3D equivalent: it takes as input a 3D volume or a sequence of 2D frames (e.g. slices in a CT scan), 3D CNNs are a powerful model for learning representations for volumetric data. It uses a subset of the [MosMedData: Chest CT Scans with COVID-19 Related Findings](https://arxiv.org/abs/2005.06465). This dataset consists of lung CT scans with COVID-19 related findings, as well as without such findings.\n\nFor more information please read the [original tutorial here](https://keras.io/examples/vision/3D_image_classification/)\n\n\n## Running the Code\n\n```\npip install -r requirements.txt\npython train.py --max_epochs 3\n```\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Faribornstein%2Fkeras-3d-image-classification","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Faribornstein%2Fkeras-3d-image-classification","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Faribornstein%2Fkeras-3d-image-classification/lists"}