Productive Reproducible Workflows for DNNs: A Case Study for Industrial Defect Detection

Gibson, P. and Cano, J. (2022) Productive Reproducible Workflows for DNNs: A Case Study for Industrial Defect Detection. 4th Workshop on Accelerated Machine Learning (AccML), co-located with the HiPEAC conference, Budapest, 22 June 2022.

[img] Text
280376.pdf - Accepted Version


Publisher's URL:


As Deep Neural Networks (DNNs) have become an increasingly ubiquitous workload, the range of libraries and tooling available to aid in their development and deployment has grown significantly. Scalable, production quality tools are freely available under permissive licenses, and are accessible enough to enable even small teams to be very productive. However within the research community, awareness and usage of said tools is not necessarily widespread, and researchers may be missing out on potential productivity gains from exploiting the latest tools and workflows. This paper presents a case study where we discuss our recent experience producing an end-to-end artificial intelligence application for industrial defect detection. We detail the high level deep learning libraries, containerized workflows, continuous integration/deployment pipelines, and open source code templates we leveraged to produce a competitive result, matching the performance of other ranked solutions to our three target datasets. We highlight the value that exploiting such systems can bring, even for research, and detail our solution and present our best results in terms of accuracy and inference time on a server class GPU, as well as inference times on a server class CPU, and a Raspberry Pi 4.

Item Type:Conference or Workshop Item
Keywords:deep learning, docker, defect detection, pytorch, reproducibility, bonseyes
Glasgow Author(s) Enlighten ID:Gibson, Perry and Cano Reyes, Dr Jose
Authors: Gibson, P., and Cano, J.
College/School:College of Science and Engineering > School of Computing Science
Copyright Holders:Copyright © 2022 The Authors
Publisher Policy:Reproduced with the permission of the Author
Related URLs:

University Staff: Request a correction | Enlighten Editors: Update this record