利用Jetson NANO做一个急性淋巴细胞白血病分类器

2021-09-22 11:25:15 浏览数 (1)

急性淋巴细胞白血病(ALL)是一种起源于淋巴细胞的B系或T系细胞在骨髓内异常增生的恶性肿瘤性疾病。异常增生的原始细胞可在骨髓聚集并抑制正常造血功能,同时也可侵及骨髓外的组织,如脑膜、淋巴结、性腺、肝等。

急性淋巴细胞白血病 Jetson Nano 分类器是一种卷积神经网络 (CNN),它使用英特尔® oneAPI AI 分析工具包和英特尔® Tensorflow* 优化来加速训练过程,并使用 TensorRT 在 NVIDIA® Jetson Nano™ 上进行高性能推理。

在本项目中,您将执行以下操作。

  • 在您的开发机器上训练用于急性淋巴细胞白血病的定制 CNN。
  • 将 Tensorflow SavedModel 转换为 TFRT 格式
  • 将 Tensorflow SavedModel 转换为 ONNX 格式
  • 将 Tensorflow ONNX 模型转换为 TensorRT 格式
  • 在您的开发机器上测试 CNN。
  • 为您的 NVIDIA Jetson Nano 下载或创建自定义 docker 容器。
  • 使用 Tensorflow 在 NVIDIA Jetson Nano 上运行 CNN。
  • 使用 TFRT 在 NVIDIA Jetson Nano 上运行 CNN。
  • 使用 TensorRT 在 NVIDIA Jetson Nano 上运行 CNN。

NVIDIA Jetson Nano

NVIDIA® Jetson Nano™ 是一款功能强大的小型计算机,能够并行运行多个神经网络,用于图像分类、对象检测、分割和语音处理等应用。

TFRT

TFRT 是一个新的 TensorFlow 运行时。它旨在提供一个统一的、可扩展的基础设施层,在各种特定领域的硬件上具有一流的性能。它提供多线程主机 CPU 的高效使用,支持完全异步编程模型,以及专注于低水平的效率。

ONNX

ONNX 是一种开放格式,用于表示机器学习模型。ONNX 定义了一组通用运算符——机器学习和深度学习模型的构建块——以及一种通用文件格式,使 AI 开发人员能够使用具有各种框架的模型,工具、运行时和编译器。

TensorRT

“NVIDIA® TensorRT™ 是用于高性能深度学习推理的 SDK。它包括深度学习推理优化器和运行时,可为深度学习推理应用程序提供低延迟和高吞吐量。”

开源代码:

https://github.com/AMLResearchProject/ALL-Jetson-Nano-Classifier

0 人点赞