awoo
Browse filesSigned-off-by: Balazs Horvath <[email protected]>
README.md
CHANGED
@@ -290,11 +290,13 @@ pip install timm
|
|
290 |
|
291 |
The following inference script for the tagger needs a folder as input, be warned that it also converts WebP images to PNG and you can specify tags to be ignored and some other stuff! I recommend reading through it and changing whatever you need.
|
292 |
|
|
|
|
|
293 |
<div style="background-color: lightyellow; padding: 10px;">
|
294 |
<details>
|
295 |
<summary>Click to reveal inference script</summary>
|
296 |
|
297 |
-
```
|
298 |
import os
|
299 |
import torch
|
300 |
from torchvision import transforms
|
|
|
290 |
|
291 |
The following inference script for the tagger needs a folder as input, be warned that it also converts WebP images to PNG and you can specify tags to be ignored and some other stuff! I recommend reading through it and changing whatever you need.
|
292 |
|
293 |
+
[Colab Notebook](https://colab.research.google.com/drive/1gIB2fGjLAuh6s_hrNlIPCkw_3jodoFP0?usp=sharing)
|
294 |
+
|
295 |
<div style="background-color: lightyellow; padding: 10px;">
|
296 |
<details>
|
297 |
<summary>Click to reveal inference script</summary>
|
298 |
|
299 |
+
```python
|
300 |
import os
|
301 |
import torch
|
302 |
from torchvision import transforms
|