self-use toolkits
Project description
kiui kit
A toolkit for personal use.
Install
# released
pip install kiui # install the minimal package
pip install kiui[full] # install optional dependencies
# latest
pip install git+https://github.com/ashawkey/kiuikit.git # only the minimal package
Usage
import kiui
### auto import
kiui.env() # os, glob, math, time, random, argparse
kiui.env('data') # above + np, plt, cv2, Image, ...
kiui.env('torch') # above + torch, nn, F, ...
### quick inspection of array-like object
x = torch.tensor(...)
y = np.array(...)
kiui.lo(x)
kiui.lo(x, y) # support multiple objects
kiui.lo(kiui) # or any other object (just print with name)
### io utils
# read image as-is in RGB order
img = kiui.read_image('image.png', mode='float') # mode: float (default), pil, uint8, tensor
# write image
kiui.write_image('image.png', img)
### visualization tools
img_tensor = torch.rand(3, 256, 256)
# tensor of [3, H, W], [1, H, W], [H, W] / array of [H, W ,3], [H, W, 1], [H, W] in [0, 1]
kiui.vis.plot_image(img)
kiui.vis.plot_image(img_tensor)
### mesh utils
from kiui.mesh import Mesh
mesh = Mesh.load('model.obj')
kiui.lo(mesh.v, mesh.f) # CUDA torch.Tensor suitable for nvdiffrast
mesh.write('new.obj')
### background removal utils
from kiui.bg import remove, remove_file, remove_folder
res = remove(img)
remove_file('input.jpg', 'output.png')
remove_file('input.jpg', 'output.png', post_process=True) # morphology opening
remove_file('input.jpg', 'output.png', lcc=True) # largest connected component
remove_file('input.jpg', 'mask.png', return_mask=True) # only save [h, w] mask
remove_folder('input/', 'output/')
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
kiui-0.1.5.tar.gz
(14.6 kB
view hashes)
Built Distribution
kiui-0.1.5-py3-none-any.whl
(15.4 kB
view hashes)