2 个版本
0.1.1 | 2022年2月12日 |
---|---|
0.1.0 | 2022年2月12日 |
#54 in #tensorflow
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"
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