Machine Learning with Selenium and TensorFlow

Table of Contents

• About the Project
  • Built With
• Getting Started
  • Prerequisites
  • Installation
• Usage
• Contributing
• License
• Contact
• Acknowledgements

About The Project

This project is based on this repository seleniumMachineLearning

Train a Tensorflow model based on Inception V3 with custom set of images and apply the model to identify objects in a webpage with the help of selenium.

Built With

• TensorFlow
• Docker
• Selenium
• And a bunch of python libraries

Getting Started

This is an example of how you may give instructions on setting up your project locally. To get a local copy up and running follow these simple example steps.

Prerequisites

This is the list of things you need to have in your system.

1. Get Docker Desktop to pull Tensorflow images
2. Have Python installed in the system

Installation

1. Clone the repo

git clone https://github.com/dasxran/seleniumMachineLearning.git

2. Pull Tensorflow image from Docker hub

docker pull dasxran/tensorflow:trainimages

Usage

python3 tf_exec_identifySearchButton.py

Note that the chromedriver location will likely need to be changed, and the paths are linux style.

License

Distributed under the MIT License. See LICENSE for more information.

Acknowledgements

Based on work by Ranjan Kumar Das Project Link: https://github.com/dasxran/seleniumMachineLearning