Logo
Logo
Desktop
By Muhammad Abubakar4 min read

HeatGen: Deep Learning Heatmap Prediction Tool– Backed by Published Research

I led the research, published it in a peer-reviewed journal, and then built a desktop app around it, including GPU-accelerated training with no coding required to use it.

HeatGen: Deep Learning Heatmap Prediction Tool– Backed by Published Research

Technologies Used

PythonPython
FlaskFlask
QTQT

Overview

HeatGen is a desktop application that I designed and built to convert raw datasets into heatmaps and run deep learning-based analysis– for surfacing patterns, anomalies, and correlations automatically through a point-and-click interface. It is directly grounded in original research I led as primary author, peer-reviewed, and published in the Journal of Engineering Science and Technology in April 2023.

The project produced two parallel outputs: a published research paper validating the deep learning approach, and a functional desktop tool that makes the same pipeline accessible without a technical background.

The Research

I investigated whether a deep learning model could accurately predict the surface temperature profiles of pavement bricks embedded with phase change materials (PCMs)– without requiring a physical experiment each time. PCMs are being actively studied as a passive cooling solution to the Urban Heat Island effect in dense, high-temperature urban environments.

I selected three PCMs– lauric acid, stearic acid, and paraffin wax– and conducted physical experiments outdoors in Cheras, Selangor, Malaysia, using a thermal imaging camera to capture heatmaps of each brick sample. From this, I built a machine learning dataset associating 11 physical variables with their resulting thermal heatmaps, applied a 90/10 train-test split, and designed a deep learning model using fully connected and convolutional layers. The model achieved a mean squared error of 0.0465 on unseen test data– confirming its ability to generalise across different PCM formulations and temperature profiles.

The Problem

During the research, I noticed a practical gap. The lead researcher was an academic domain expert– his focus was on science, not the software, which made sense. The side effect was that once the model worked, there was no way to actually use it outside of a lab. More broadly, anyone working with thermal or high-dimensional datasets faced the same wall:

No Heatmap Generation

No way to generate 2D color-coded heatmaps directly from raw data without writing custom code.

No Execution Environment

No accessible environment available to run deep learning analysis pipelines efficiently.

Fragmented Tooling

No unified tool that combines training, testing, visualization, and results in one place.

I built HeatGen to close that gap– for this research, and for anyone who might need the same capability without the technical overhead.

My Approach

01
Backend in Python

For the backend, I chose Python because it’s the standard language for the data science and machine learning ecosystem, and I knew this would let future researchers build on my work.

02
GPU-Accelerated Model Training

Training the model on CPU took approximately 4 hours per run– far too slow for iterative experimentation. So I integrated CUDA with an NVIDIA GPU, which reduced training time to around 15 minutes. This single change made proper hyperparameter tuning and iteration actually feasible for my use case.

03
Dataset Preprocessing

I preprocessed the image dataset by resizing the thermal heatmap images to a consistent dimension (160×120 pixels) to match the model’s input requirements, and normalized all variable values between 0 and 1. I then applied a 90/10 train-test split to evaluate the model’s generalization on unseen data.

04
Training and Testing Tool

Rather than hardcoding a single model into the application, I built a dedicated training and testing module within HeatGen. This means the app is not locked to one version of the model– as more experimental data is collected, the model can be retrained and fine-tuned directly within the tool, without rebuilding the application.

05
PyQt5 Desktop Interface

built the frontend using PyQt5 to give HeatGen a proper desktop UI– this supported file uploads, model selection, heatmap visualisation, and results output, all accessible through a point-and-click interface with no command line knowledge required.

06
Multithreading for UI Stability

Model training can run for several minutes, so I added multithreading to handle it in the background. This way, the interface stays usable even while a job is running rather than locking up until it finishes, and the application remains responsive throughout.

What I Built

HeatGen is a fully functional desktop application with the following core capabilities:

  • Automated heatmap generation– transforms raw input data into 2D colour-gradient visualizations, and maps values to colour ranges for immediate pattern recognition
  • Integrated deep learning pipeline– the same model architecture validated in the published research, embedded directly into my tool for anomaly detection, clustering, and pattern identification
  • Training and fine-tuning module– users can retrain the model on new data as it becomes available, so that the tool remains useful beyond the original dataset
  • GPU-accelerated processing– CUDA integration reduces model training from hours to minutes
  • Accessible Qt interface– designed for both technical and non-technical users, with no scripts, terminal access, or manual configuration required
  • Flexible input and output– standard dataset formats are supported, with clean i and summary results

Results

Metric
Model accuracy (MSE)
Result
0.0465 on unseen test data
Metric
Training time (CPU)
Result
~4 hours
Metric
Training time (GPU with CUDA)
Result
~15 minutes
Metric
Publication
Result
JESTEC Vol. 18, Issue 2, April 2023
Metric
Dataset novelty
Result
First published dataset and DL model for PCM heatmap prediction

My model demonstrated consistent generalization across multiple PCM formulations and temperature profiles. The research was peer-reviewed and accepted for publication, and to my knowledge, the first time a dataset and deep learning model of this kind had been developed specifically for predicting heatmaps from PCM-related physical variables.

Need a developer who can take a project from research to production?

I work across the full stack– from data and modeling to shipping tools that your team can actually use. Let’s talk!

Questions? Comments? Feel free to send a message!