tensorneko 0.3.21__py3-none-any.whl → 0.3.22__py3-none-any.whl

tensorneko/evaluation/psnr.pyi

@@ -0,0 +1,21 @@
+from typing import overload
+
+from torch import Tensor
+
+from .enum import Reduction
+
+
+@overload
+def psnr_image(pred: str, real: str) -> Tensor: ...
+
+
+@overload
+def psnr_image(pred: Tensor, real: Tensor, reduction: Reduction = Reduction.MEAN) -> Tensor: ...
+
+
+@overload
+def psnr_video(pred: str, real: str, use_ffmpeg: bool = False) -> Tensor: ...
+
+
+@overload
+def psnr_video(pred: Tensor, real: Tensor) -> Tensor: ...

tensorneko/evaluation/ssim.pyi

@@ -0,0 +1,21 @@
+from typing import overload
+
+from torch import Tensor
+
+from .enum import Reduction
+
+
+@overload
+def ssim_image(pred: str, real: str) -> Tensor: ...
+
+
+@overload
+def ssim_image(pred: Tensor, real: Tensor, reduction: Reduction = Reduction.MEAN) -> Tensor: ...
+
+
+@overload
+def ssim_video(pred: str, real: str, use_ffmpeg: bool = False) -> Tensor: ...
+
+
+@overload
+def ssim_video(pred: Tensor, real: Tensor) -> Tensor: ...

tensorneko/preprocess/crop.pyi

@@ -0,0 +1,16 @@
+from typing import overload, Union
+
+from numpy import ndarray
+from torch import Tensor
+
+
+@overload
+def crop_with_padding(image: ndarray, x1: int, x2: int, y1: int, y2: int, pad_value: Union[int, float] = 0.,
+    batch: bool = False
+) -> ndarray: ...
+
+
+@overload
+def crop_with_padding(image: Tensor, x1: int, x2: int, y1: int, y2: int, pad_value: Union[int, float] = 0.,
+    batch: bool = False
+) -> Tensor: ...

tensorneko/util/dispatched_misc.pyi

@@ -0,0 +1,34 @@
+from typing import overload, List
+
+from numpy import ndarray
+from torch import Tensor
+
+
+@overload
+def sparse2binary(x: Tensor, length: int = None) -> Tensor:
+    ...
+
+
+@overload
+def sparse2binary(x: ndarray, length: int = None) -> ndarray:
+    ...
+
+
+@overload
+def sparse2binary(x: List[int], length: int = None) -> ndarray:
+    ...
+
+
+@overload
+def binary2sparse(x: Tensor) -> Tensor:
+    ...
+
+
+@overload
+def binary2sparse(x: ndarray) -> ndarray:
+    ...
+
+
+@overload
+def binary2sparse(x: List[int]) -> List[int]:
+    ...

tensorneko/version.txt

@@ -1 +1 @@
-0.3.21
+0.3.22

{tensorneko-0.3.21.dist-info → tensorneko-0.3.22.dist-info}/METADATA

@@ -1,21 +1,20 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: tensorneko
-Version: 0.3.21
+Version: 0.3.22
 Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
 Author-email: smczx@hotmail.com
-License: UNKNOWN
 Project-URL: Bug Tracker, https://github.com/ControlNet/tensorneko/issues
 Project-URL: Source Code, https://github.com/ControlNet/tensorneko
 Keywords: deep learning,pytorch,AI,data processing
-Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Intended Audience :: Developers
@@ -23,19 +22,33 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Utilities
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: av (>=8.0.3)
-Requires-Dist: einops (>=0.3.0)
-Requires-Dist: numpy (>=1.20.1)
-Requires-Dist: pillow (>=8.1)
-Requires-Dist: tensorneko-util (==0.3.21)
-Requires-Dist: torch (>=1.9.0)
-Requires-Dist: torchaudio (>=0.9.0)
-Requires-Dist: torchmetrics (>=0.7.3)
-Requires-Dist: torchvision (>=0.10.0)
-Requires-Dist: pysoundfile (>=0.9.0) ; platform_system == "Windows"
+License-File: LICENSE
+Requires-Dist: torch>=1.9.0
+Requires-Dist: torchaudio>=0.9.0
+Requires-Dist: torchvision>=0.10.0
+Requires-Dist: torchmetrics>=0.7.3
+Requires-Dist: pillow>=8.1
+Requires-Dist: av>=8.0.3
+Requires-Dist: pysoundfile>=0.9.0; platform_system == "Windows"
+Requires-Dist: numpy>=1.20.1
+Requires-Dist: einops>=0.3.0
+Requires-Dist: tensorneko_util==0.3.22
 Provides-Extra: lightning
-Requires-Dist: lightning (<3,>=2.0) ; extra == 'lightning'
-Requires-Dist: tensorboard (>=2.0.0) ; extra == 'lightning'
+Requires-Dist: tensorboard>=2.0.0; extra == "lightning"
+Requires-Dist: lightning<3,>=2.0; extra == "lightning"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 <h1 style="text-align: center">TensorNeko</h1> 
 
@@ -71,7 +84,7 @@ pip install tensorneko  # for PyTorch only
 pip install tensorneko[lightning]  # for PyTorch and Lightning
 ```
 
-To use the library without PyTorch and PyTorch Lightning, you can install the util library (support Python 3.7 ~ 3.12 with limited features) with following command.
+To use the library without PyTorch and PyTorch Lightning, you can install the util library (support Python 3.7 ~ 3.13 with limited features) with following command.
 ```shell
 pip install tensorneko_util
 ```
@@ -276,7 +289,7 @@ t0 = time.time()
 with Server(view, port=8000):
     for i, data in enumerate(data_list):
         preprocessing(data) # do some processing here
-
+        
         x = time.time() - t0  # time since the start of the program
         y = i # processed number of data
         line_chart.add(x, y)  # add to the line chart
@@ -525,7 +538,7 @@ def process_data(n: int):
         return n
     else:
         return None
-
+    
 
 data = get_data()
 data = data.map(process_data).get_or_else(-1)  # if the response is None, return -1
@@ -734,5 +747,3 @@ The `gotify` can send a message to the Gotify server, with the environment varia
 ```shell
 tensorneko gotify "Script finished!"
 ```
-
-

{tensorneko-0.3.21.dist-info → tensorneko-0.3.22.dist-info}/RECORD

@@ -2,7 +2,7 @@ tensorneko/__init__.py,sha256=uh1HNn1sNpX1bbOqAE_kNJfrH4eMtEzus0hO-Fh9tEw,990
 tensorneko/neko_model.py,sha256=hUMi7puzxW_6FOpA1jiFN1__oO5DZPlhhp3WXqhJXgg,10581
 tensorneko/neko_module.py,sha256=qELXvguSjWo_NvcRQibiFl0Qauzd9JWLSnT4dbGNS3Y,1473
 tensorneko/neko_trainer.py,sha256=GqdRsPkzWj36DJ_Wroe1TI6QKlk3N8Q8gK4uEnI0w9Q,10190
-tensorneko/version.txt,sha256=UfuS4cRkfbFuN1IeX9oKpvyZc11Pi5MXqTuYUmK9f5I,7
+tensorneko/version.txt,sha256=6OtxZVjxPBSh3JDaCbhkG17yxSVMldJs7t7a1E9-Q58,7
 tensorneko/arch/__init__.py,sha256=w4lTUeyBIZelrnSjlBFWUF0erzOmBFl9FqeWQuSOyKs,248
 tensorneko/arch/auto_encoder.py,sha256=j6PWWyaNYaYNtw_zZ9ikzhCASqe9viXR3JGBIXSK92Y,2137
 tensorneko/arch/binary_classifier.py,sha256=1MkEbReXKLdDksRG5Rsife40grJk08EVDcNKp54Xvb4,2316
@@ -31,8 +31,10 @@ tensorneko/evaluation/enum.py,sha256=s3P8XAobku-as4in5vh6BanvVW5Ccwnff0t124lVFFg
 tensorneko/evaluation/fid.py,sha256=fNuE1CEp2rPXbaZfI0E1CspluInzFlUdKc8XZEexUME,5568
 tensorneko/evaluation/iou.py,sha256=phEmOWQ3cnWW377WeSHCoB8mGkHLHMHCl8_LL0IX3JA,2914
 tensorneko/evaluation/psnr.py,sha256=DeKxvY_xxawWMXHY0z3Nvbsi4dR57OUV4hjtUoCINXc,3757
+tensorneko/evaluation/psnr.pyi,sha256=aRGK9JEyXGyK3Wc247xFiajocOP-ukvvXub6yUji2CQ,416
 tensorneko/evaluation/secs.py,sha256=D710GgcSxQgbGyPcWlC5ffF5n1GselLrUr5aA5Vq7oE,1622
 tensorneko/evaluation/ssim.py,sha256=6vPS4VQqoKxHOG49lChH51KxwNo07B4XHdhLub5DEPU,3758
+tensorneko/evaluation/ssim.pyi,sha256=I5vGo2KiPajl-z_VRwAHVLF4UIrrEKBdyiY2Gq3NOJw,416
 tensorneko/io/__init__.py,sha256=QEyA0mOC-BlKKskYYbDYttYWWRjCeh73lX-yKAUGNik,213
 tensorneko/io/reader.py,sha256=DSeTGLh84sFYwCwJmNTr-fGWkluudCbf7je29t0Z2U8,1303
 tensorneko/io/writer.py,sha256=BR_1h-wXekBdctXymJBU44HoWsKxPhbbh6N3AKYNkjE,1292
@@ -69,6 +71,7 @@ tensorneko/optim/__init__.py,sha256=89XjYQICij8SkrW5iryfZgmbxcTDxA3hhVZTgR4588o,
 tensorneko/optim/lr_scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tensorneko/preprocess/__init__.py,sha256=0Z0eA3_I2wphyyZlzZYRrx2muWTF0QMFq2Y-jh8oVKU,808
 tensorneko/preprocess/crop.py,sha256=Y9eWyYdzasQK3US2uBP_sUO9bPVedik0pnrrl006zZ4,2732
+tensorneko/preprocess/crop.pyi,sha256=Aq_ywCG4ovcopH5-sj8ZVLchLgfpnt0z9cvSPccWHn8,411
 tensorneko/preprocess/enum.py,sha256=Wp5qFaUjea5XU4o3N0WxUd-qfzI-m5vr4ZWSqWjELb4,874
 tensorneko/preprocess/pad.py,sha256=b4IbbhGNRotZ7weZcKA7hfDqSixPo5KjM6khnqzaeUA,3238
 tensorneko/preprocess/resize.py,sha256=hitMlzVnN6n_8nEJwxy4C4ErZrTwpM86QGnYewsrmf8,3469
@@ -76,6 +79,7 @@ tensorneko/preprocess/face_detector/__init__.py,sha256=_ktIfUZqGTX0hk7RBgKf-zHwG
 tensorneko/util/__init__.py,sha256=39G34a2k5ktVtBAh4N4RMsePEak5rzPDhbNNdXo-Ye4,2258
 tensorneko/util/configuration.py,sha256=xXeAjDh1FCNTmSPwDdkL-uH-ULfzFF6Fg0LT7gsZ6nQ,2510
 tensorneko/util/dispatched_misc.py,sha256=_0Go7XezdYB7bpMnCs1MDD_6mPNoWP5qt8DoKuPxynI,997
+tensorneko/util/dispatched_misc.pyi,sha256=K8qZehCayr-nQIifK0w2Kp6pjVQ9uiJ0ZvxF7G5VxPA,520
 tensorneko/util/gc.py,sha256=P3bOZ-2VUNyswnfVz5xfj__ecTSAHpu_kLp2wFcpb6M,185
 tensorneko/util/misc.py,sha256=LEvACtGDOX43iK86A8-Cek0S9rbXFR0AtTP1edE3XDI,4701
 tensorneko/util/reproducibility.py,sha256=sw1vVi7VOnmzQYUocI5x9yKeZoHHiA4A5ja136XolrI,2102
@@ -86,8 +90,8 @@ tensorneko/visualization/log_graph.py,sha256=NvOwWVc_petXWYdgaHosPFLa43sHBeacbYc
 tensorneko/visualization/matplotlib.py,sha256=xs9Ssc44ojZX65QU8-fftA7Ug_pBuZ3TBtM8vETNq9w,1568
 tensorneko/visualization/image_browser/__init__.py,sha256=AtykhAE3bXQS6SOWbeYFeeUE9ts9XOFMvrL31z0LoMg,63
 tensorneko/visualization/watcher/__init__.py,sha256=Nq752qIYvfRUZ8VctKQRSqhxh5KmFbWcqPfZlijVx6s,379
-tensorneko-0.3.21.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
-tensorneko-0.3.21.dist-info/METADATA,sha256=Wc-2VJsFvNoT8f5d4rliimhmpxRAug5aVBmOL5TahQY,20813
-tensorneko-0.3.21.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
-tensorneko-0.3.21.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
-tensorneko-0.3.21.dist-info/RECORD,,
+tensorneko-0.3.22.dist-info/licenses/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
+tensorneko-0.3.22.dist-info/METADATA,sha256=B0Hzs1YnIRVHChwNAabMoYfpQ1NoyjuYmiRpsRNDHGc,21104
+tensorneko-0.3.22.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+tensorneko-0.3.22.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
+tensorneko-0.3.22.dist-info/RECORD,,

{tensorneko-0.3.21.dist-info → tensorneko-0.3.22.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.34.2)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any