Category: Tensorflow
-
How to resize an image with Tensorflow tf.image.resize()
When dealing with training data, you often need to resize images to a fixed size, according to the architecture of the machine learning model that you want to use. In Tensorflow we can use tf.image.resize() to resize images to different resolutions. The method controls the different algorithms that we can use for resizing. You can […]
-
Tensorflow not detecting GPU
If TensorFlow doesn’t detect your GPU, it will default to the CPU, which means when doing heavy training jobs, these will take a really long time to complete. This is most likely because the CUDA and CuDNN drivers are not being correctly detected in your system. I am assuming that you have already installed Tensorflow […]
-
Does CUDA 11.6 work with Tensorflow?
The short answer is No. The latest Tensorflow, which is currently Tensorflow 2.9, is not yet ready to work with CUDA 11.6. If you check the Tensorflow compatibility matrix below you will find that as of today the recommended CUDA version you should install is 11.2: Version Python version Compiler Build tools cuDNN CUDA tensorflow-2.9.0 […]
-
Installing Tensorflow 2.9 with GPU Support
You have come here because you want to install Tensorflow 2.9, and you want to get it right for the first time. Typically installing Tensorflow is hard work due to the number of dependencies. The good news is that since Tensorflow 2.5, the basic dependencies haven’t changed much, CUDA 11.2 and CuDNN 8.1 is still […]
-
How to setup Tensorflow on Ubuntu Linux with multiple GPUs using Docker
For this tutorial, I will be setting up the latest version of Tensorflow, currently 2.9 with GPU support on a workstation kindly provided by SabrePC with 4 x RTX 3080. For those curious, here is a clip of what this workstation looks like: Setting up Tensorflow typically requires a lot of hard work as you […]
-
How to crop and resize an image using Tensorflow.js tf.image.cropAndResize
This is a quick post to show how you can use tf.image.cropAndResize to crop and resize an image to a square using the function tf.image.cropAndResize. This function was created for a specific purpose to cutout the bounding boxes given by a face detection algorithm. But you can use it for other purposes too! Your image […]
-
A tutorial on how to convert a Tensorflow model to Tensorflow.js
In this article, I will take you through the steps that are needed to convert any model in Tensorflow Hub to Tensorflow.JS format, and how you can quickly make a simple interactive page that you can host on any web server on the internet. I am only scratching the surface of what is possible. More […]
-
Real-time Fast Style Transfer model with Tensorflow.js and Next.js in the Browser
I wanted to share a project that I have been working on using Next.js and React. My challenge was to create an interactive component using React/Next.JS that anyone could try from a browser to play with the Fast Style Transfer model in real-time, via a web camera all within the browser. Let’s just say that […]
-
Creating my own image dataset for Tensorflow with CVAT
So, I have created a Raspberry PI wildlife camera, but it has a big cat issue! Or shall I say, a big dog issue?! The Tensorflow model I am using mistakes everything for a dog. A fox, a cat, a person, and some sunshine can all be mistaken for a dog. So I have to do […]
-
Flutter Image classification using TensorFlow in 4 steps
Google provides a few products within the TensorFlow family: TensorFlow: the core open course library that is the foundation of developing and training machine learning models. TensorFlow.js: similar to TensorFlow but focus purely on JavaScript TensorFlow Lite: as the name suggests, it is a lightweight version of TensorFlow for deploying models on mobile devices. It […]
