Skip to content

Commit a2b1993

Browse files
authored
Merge pull request #7 from DefTruth/dev
fix(setup): update pypi v0.1.0 (#6)
2 parents e9f2f1a + 0cf93eb commit a2b1993

File tree

2 files changed

+3
-11
lines changed

2 files changed

+3
-11
lines changed

README.md

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212

1313
## 🤗 Introduction
14-
**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**.
14+
**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**.
1515

1616
# 🆕 What's New
1717

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

98-
9998
def callable_tensor_noop(
10099
img: Tensor,
101100
landmarks: Tensor
@@ -152,31 +151,25 @@ torchlm.set_transforms_debug(True)
152151
torchlm.set_transforms_logging(True)
153152
torchlm.set_autodtype_logging(True)
154153
```
155-
Some details logs will show you at each runtime, just like the follows
154+
some detail information will show you at each runtime, the infos might look like
156155
```shell
157-
LandmarksRandomHorizontalFlip() AutoDtype Info: AutoDtypeEnum.Array_InOut
158-
LandmarksRandomHorizontalFlip() Execution Flag: True
159156
LandmarksRandomScale() AutoDtype Info: AutoDtypeEnum.Array_InOut
160157
LandmarksRandomScale() Execution Flag: False
161-
...
162158
BindTorchVisionTransform(GaussianBlur())() AutoDtype Info: AutoDtypeEnum.Tensor_InOut
163159
BindTorchVisionTransform(GaussianBlur())() Execution Flag: True
164-
...
165160
BindAlbumentationsTransform(ColorJitter())() AutoDtype Info: AutoDtypeEnum.Array_InOut
166161
BindAlbumentationsTransform(ColorJitter())() Execution Flag: True
167-
...
168162
BindArrayCallable(callable_array_noop())() AutoDtype Info: AutoDtypeEnum.Array_InOut
169163
BindArrayCallable(callable_array_noop())() Execution Flag: True
170164
BindTensorCallable(callable_tensor_noop())() AutoDtype Info: AutoDtypeEnum.Tensor_InOut
171165
BindTensorCallable(callable_tensor_noop())() Execution Flag: True
172-
...
173166
LandmarksUnNormalize() AutoDtype Info: AutoDtypeEnum.Array_InOut
174167
LandmarksUnNormalize() Execution Flag: True
175168
```
176169
* 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).
177170
* AutoDtype Info:
178171
* Array_InOut means current transform need a np.ndnarray as input and then output a np.ndarray.
179-
* Tensor_InOut means current transform need a torch Tensor as input and then output torch Tensor.
172+
* Tensor_InOut means current transform need a torch Tensor as input and then output a torch Tensor.
180173
* Array_In means current transform needs a np.ndarray input and then output a torch Tensor.
181174
* Tensor_In means current transform needs a torch Tensor input and then output a np.ndarray.
182175

test/transforms.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@ def test_torchlm_transforms():
4848

4949
transform = torchlm.LandmarksCompose([
5050
# use native torchlm transforms
51-
torchlm.LandmarksRandomHorizontalFlip(prob=0.5),
5251
torchlm.LandmarksRandomScale(prob=0.5),
5352
torchlm.LandmarksRandomTranslate(prob=0.5),
5453
torchlm.LandmarksRandomShear(prob=0.5),

0 commit comments

Comments
 (0)