Skip to content

Commit

Permalink
Merge pull request #7 from DefTruth/dev
Browse files Browse the repository at this point in the history
fix(setup): update pypi v0.1.0 (#6)
  • Loading branch information
DefTruth authored Feb 13, 2022
2 parents e9f2f1a + 0cf93eb commit a2b1993
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 11 deletions.
13 changes: 3 additions & 10 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@


## 🤗 Introduction
**torchlm** is a PyTorch landmarks-only library with **100+ data augmentations**, **training** and **inference**. **torchlm** is only focus on any landmarks detection, such as face landmarks, hand keypoints and body keypoints, etc. It provides **30+** native data augmentations and compatible with **80+** torchvision and albumations's transforms, no matter the input is a np.ndarray or a torch Tensor, **torchlm** will **automatically** be compatible with different data types through a **autodtype** wrapper. Further, in the future **torchlm** will add modules for **training** and **inference**.
**torchlm** is a PyTorch landmarks-only library with **100+ data augmentations**, **training** and **inference**. **torchlm** is only focus on any landmarks detection, such as face landmarks, hand keypoints and body keypoints, etc. It provides **30+** native data augmentations and compatible with **80+** torchvision and albumations's transforms, no matter the input is a np.ndarray or a torch Tensor, **torchlm** will automatically be compatible with different data types and then wrap back to the original type through a autodtype wrapper. Further, in the future **torchlm** will add modules for **training** and **inference**.

# 🆕 What's New

Expand Down Expand Up @@ -95,7 +95,6 @@ def callable_array_noop(
# Do some transform here ...
return img.astype(np.uint32), landmarks.astype(np.float32)


def callable_tensor_noop(
img: Tensor,
landmarks: Tensor
Expand Down Expand Up @@ -152,31 +151,25 @@ torchlm.set_transforms_debug(True)
torchlm.set_transforms_logging(True)
torchlm.set_autodtype_logging(True)
```
Some details logs will show you at each runtime, just like the follows
some detail information will show you at each runtime, the infos might look like
```shell
LandmarksRandomHorizontalFlip() AutoDtype Info: AutoDtypeEnum.Array_InOut
LandmarksRandomHorizontalFlip() Execution Flag: True
LandmarksRandomScale() AutoDtype Info: AutoDtypeEnum.Array_InOut
LandmarksRandomScale() Execution Flag: False
...
BindTorchVisionTransform(GaussianBlur())() AutoDtype Info: AutoDtypeEnum.Tensor_InOut
BindTorchVisionTransform(GaussianBlur())() Execution Flag: True
...
BindAlbumentationsTransform(ColorJitter())() AutoDtype Info: AutoDtypeEnum.Array_InOut
BindAlbumentationsTransform(ColorJitter())() Execution Flag: True
...
BindArrayCallable(callable_array_noop())() AutoDtype Info: AutoDtypeEnum.Array_InOut
BindArrayCallable(callable_array_noop())() Execution Flag: True
BindTensorCallable(callable_tensor_noop())() AutoDtype Info: AutoDtypeEnum.Tensor_InOut
BindTensorCallable(callable_tensor_noop())() Execution Flag: True
...
LandmarksUnNormalize() AutoDtype Info: AutoDtypeEnum.Array_InOut
LandmarksUnNormalize() Execution Flag: True
```
* Execution Flag: True means current transform was executed successful, False means it was not executed because of the random probability or some Runtime Exceptions(torchlm will should the error infos if debug mode is True).
* AutoDtype Info:
* Array_InOut means current transform need a np.ndnarray as input and then output a np.ndarray.
* Tensor_InOut means current transform need a torch Tensor as input and then output torch Tensor.
* Tensor_InOut means current transform need a torch Tensor as input and then output a torch Tensor.
* Array_In means current transform needs a np.ndarray input and then output a torch Tensor.
* Tensor_In means current transform needs a torch Tensor input and then output a np.ndarray.

Expand Down
1 change: 0 additions & 1 deletion test/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@ def test_torchlm_transforms():

transform = torchlm.LandmarksCompose([
# use native torchlm transforms
torchlm.LandmarksRandomHorizontalFlip(prob=0.5),
torchlm.LandmarksRandomScale(prob=0.5),
torchlm.LandmarksRandomTranslate(prob=0.5),
torchlm.LandmarksRandomShear(prob=0.5),
Expand Down

0 comments on commit a2b1993

Please sign in to comment.