Giter Site home page Giter Site logo

dermatologist / kedro-tf-image Goto Github PK

View Code? Open in Web Editor NEW
5.0 4.0 3.0 135 KB

Kedro pipelines for preprocessing images for TensorFlow.

Home Page: https://skinhelpdesk.com

Python 98.68% Dockerfile 1.32%
kedro tensorflow2 cnn dermatology hacktoberfest

kedro-tf-image's Introduction

Kedro TF Image ๐Ÿ–ผ๏ธ

This package consists of Kedro pipelines for preprocessing images using TensorFlow. I use it for Dermatology workflows and multimodal machine learning. Use this template that imports this package for multimodal ML. This package includes Kedro datasets for loading weights (as in CheXnet weights to a DenseNet121) and downloading pre-trained models from TF hub.

  • The download pipeline downloads online images defined in a csv file for multilabel classification. The labels are added to the filename. The csv format is:
id, url, labels
1, https://somesite.com/someimage.jpg,dog|black|grey
  • The folder pipeline creates TensorFlow dataset from a folder of images with labels as subfolders.
  • The multilabel pipeline processes files downloaded by the 'download' pipeline and create a dataset with images and labels. The labels are extracted from the filename. Example: _dog_black.jpg
  • Add labels in parameters.yml
master_labels: ["cat", "dog", "white", "black", "tan"]
val_size: 0.2

How to install

How to use

from kedro_tf_image.pipelines import preprocess

download = preprocess.create_download_pipeline(
        input="csvdata", output="imageset") #input is csv
folder = preprocess.create_folder_pipeline(
        input="imagefolder", output="processeddataset")
multilabel = preprocess.create_multilabel_pipeline(input="imageset", output="processeddataset")

# check output keys in the catalog below

imageset:
  type: PartitionedDataSet
  dataset: {
      "type": "kedro_tf_image.extras.datasets.tf_image_dataset.TfImageDataSet",
      "imagedim": 224,
      "preprocess_input": "tensorflow.keras.applications.resnet50.preprocess_input"
  }
  path: data/01_raw/imageset
  filename_suffix: ".jpg"

csvdata:
  type: pandas.CSVDataSet
  filepath: data/01_raw/csvfile.csv

imagefolder:
  type: kedro_tf_image.extras.datasets.tf_image_folder.TfImageFolder
  folderpath: "/path/to/images"
  imagedim: 224
  load_args:
    validation_split: 0.2
    seed: 123
    batch_size: 1


processeddataset:
  type: kedro_tf_image.extras.datasets.tf_image_processed.TfImageProcessed
  folderpath: data/02_intermediate/
  imagedim: 224

# This is required as copy_mode: assign is needed for TF datasets
datasetinmemory:
  type: MemoryDataSet
  copy_mode: assign

  • kedro_tf_image.extras.datasets.tf_image_dataset.TfImageDataSet - Load single images
  • kedro_tf_image.extras.datasets.tf_image_folder.TfImageFolder - Load a folder of images
  • kedro_tf_image.extras.datasets.tf_model_weights.TfModelWeights - Read model from weights (Ex: CheXnet with dim 14) (Use create_classification_layer to add a Dense layer of NCLASSES dim)
  • kedro_tf_image.extras.datasets.tf_model_download.TfModelDownload - Load model from TF hub.

Author

Overview

This is your new Kedro project, which was generated using Kedro 0.17.3.

Take a look at the Kedro documentation to get started.

Rules and guidelines

In order to get the best out of the template:

  • Don't remove any lines from the .gitignore file we provide
  • Make sure your results can be reproduced by following a data engineering convention
  • Don't commit data to your repository
  • Don't commit any credentials or your local configuration to your repository. Keep all your credentials and local configuration in conf/local/

How to install dependencies

Declare any dependencies in src/requirements.txt for pip installation and src/environment.yml for conda installation.

To install them, run:

kedro install

How to run Kedro

You can run your Kedro project with:

kedro run

How to test your Kedro project

Have a look at the file src/tests/test_run.py for instructions on how to write your tests. You can run your tests as follows:

kedro test

To configure the coverage threshold, look at the .coveragerc file.

Project dependencies

To generate or update the dependency requirements for your project:

kedro build-reqs

This will copy the contents of src/requirements.txt into a new file src/requirements.in which will be used as the source for pip-compile. You can see the output of the resolution by opening src/requirements.txt.

After this, if you'd like to update your project requirements, please update src/requirements.in and re-run kedro build-reqs.

Further information about project dependencies

How to work with Kedro and notebooks

Note: Using kedro jupyter or kedro ipython to run your notebook provides these variables in scope: context, catalog, and startup_error.

Jupyter

To use Jupyter notebooks in your Kedro project, you need to install Jupyter:

pip install jupyter

After installing Jupyter, you can start a local notebook server:

kedro jupyter notebook

JupyterLab

To use JupyterLab, you need to install it:

pip install jupyterlab

You can also start JupyterLab:

kedro jupyter lab

IPython

And if you want to run an IPython session:

kedro ipython

How to convert notebook cells to nodes in a Kedro project

You can move notebook code over into a Kedro project structure using a mixture of cell tagging and Kedro CLI commands.

By adding the node tag to a cell and running the command below, the cell's source code will be copied over to a Python file within src/<package_name>/nodes/:

kedro jupyter convert <filepath_to_my_notebook>

Note: The name of the Python file matches the name of the original notebook.

Alternatively, you may want to transform all your notebooks in one go. Run the following command to convert all notebook files found in the project root directory and under any of its sub-folders:

kedro jupyter convert --all

How to ignore notebook output cells in git

To automatically strip out all output cell contents before committing to git, you can run kedro activate-nbstripout. This will add a hook in .git/config which will run nbstripout before anything is committed to git.

Note: Your output cells will be retained locally.

Package your Kedro project

Further information about building project documentation and packaging your project

kedro-tf-image's People

Contributors

dermatologist avatar griffinw29 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

kedro-tf-image's Issues

Update README.md

๐Ÿ†•๐Ÿฅโ˜ First Timers Only.

This issue is reserved for people who never contributed to Open Source before. We know that the process of creating a pull request is the biggest barrier for new contributors. This issue is for you ๐Ÿ’

About First Timers Only.

๐Ÿค” What you need to know.

Nothing. This issue is meant to welcome you to Open Source :) We are happy to walk you through the process.

๐Ÿ“‹ Step by Step

  • ๐Ÿ™‹ Claim this issue: Comment below.

    Once claimed we add you as contributor to this repository.

  • ๐Ÿ‘Œ Accept our invitation to this repository. Once accepted, assign yourself to this issue

  • ๐Ÿ“ Update the file \README.md in the kedro-tf-image repository (press the little pen Icon) and edit the line as shown below.

@@ -1,5 +1,5 @@
 # Kedro TF Image
-This package consists of Kedro pipelines for preprocessing images for TensorFlow. I use it mostly for [CNN based Dermatology workflows.](https://skinhelpdesk.com)
+This package consists of [Kedro pipelines](https://kedro.readthedocs.io/en/stable/kedro.pipeline.html) for preprocessing images for TensorFlow. I use it mostly for [CNN based Dermatology workflows.](https://skinhelpdesk.com)
 
 * The **download** pipeline downloads online images defined in a csv file for multilabel classification. The labels are added to the filename. The csv format is:

- [ ] ๐Ÿ’พ **Commit** your changes

- [ ] ๐Ÿ”€ **Start a Pull Request**. There are two ways how you can start a pull request:

1. If you are familiar with the terminal or would like to learn it, [here is a great tutorial](https://egghead.io/series/how-to-contribute-to-an-open-source-project-on-github) on how to send a pull request using the terminal.

2. You can [edit files directly in your browser](https://help.github.com/articles/editing-files-in-your-repository/)

- [ ] ๐Ÿ **Done** Ask in comments for a review :)

### ๐Ÿค”โ“ Questions

Leave a comment below!

This issue was created by [First-Timers-Bot](https://github.com/hoodiehq/first-timers-bot).

ModuleNotFoundError: No module named 'kedro.versioning'

This Kedro project was created via Kedro 0.17.3 and has been bumped up to Kedro0.18.4 but the versioning module was removed in 0.18.x and therefore I am getting the following error:

ModuleNotFoundError: No module named 'kedro.versioning'

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.