#image #command-line-tool #rgb #grayscale #converting #input #tensorflow

bin+lib rimcol

使用 CNN 模型将灰度图转换为 RGB 图的命令行工具

2 个版本

0.1.1 2022年2月12日
0.1.0 2022年2月12日

#54 in #tensorflow

MIT 许可证

14KB
107

rimcol

rimcol - Rust 图像着色器 是一个命令行工具,用于将灰度图像转换为 RGB 图像。


rimcol 代表 rust image coloriser。它是一个用 Rust 编写的命令行工具,使用 tensorflow 和 python 编写的 CNN 模型来预测灰度图像的 RGB 色彩方案。输入图像应为 256*256 的灰度图像,输出是相同尺寸的 RGB 图像。

注意:该模型目前处于早期阶段,并且随着每次迭代而变得更加精确。


命令行选项

  • -i--path-to-input(参数):传入灰度图像的位置。
  • -o--path-for-output(参数):传入保存 RGB 图像的位置。
  • -h--help:获取帮助信息。

安装

rimcol 目前可以通过以下三种方式使用:通过 cargo 安装、从源代码构建并作为 Rust 包使用。

在安装之前,从 此处 下载模型,解压它,并将其文件夹放置在 /home/<user>/

注意:rimcol 依赖于 tensorflow 来运行 CNN 模型,因此您应该安装 tensorflow 的先决条件。请参阅 tensorflow-rust

从 GitHub 手动安装

当发布版本时,rimcol 的编译二进制版本会被上传到 GitHub。您可以通过 下载发布版,解压它,并将二进制文件复制到您的 $PATH 目录中,例如 /usr/local/bin

Cargo

如果您已经设置了 Rust 环境,您可以使用 cargo install 命令

$ cargo install rimcol

Cargo 将构建 rimcol 二进制文件并将其放置在 $HOME/.cargo

作为一个包

如果您想在 Rust 项目中使用 rimcol 作为包,请将以下内容添加到您的 Cargo.toml 文件中

rimcol = "0.1.0"

Rust 1.57.0+ MIT Licence

rimcol是用Rust编写的。您需要rustc版本1.57.0或更高版本。开发Rust的推荐安装方式是从官方下载页面,使用rustup。

Rust安装完成后,您可以使用Cargo编译从GitHub克隆的rimcol目录。

$ cargo build
  • 如果您想编译自己的版本,请运行以下命令:cargo build --release

将生成的二进制文件复制到您的PATH路径中的一个文件夹中,例如/usr/local/bin,二进制文件将在target/release目录中。

依赖项

~35MB
~553K SLoC