Added Ptv Image Folder Dataset node.
This commit is contained in:
@@ -275,6 +275,7 @@
|
||||
| [Ptn Linear Model](ptn_linear_model.md) | A linear model consisting of dense layers. |
|
||||
| [Ptv Dataset](ptv_dataset.md) | A Torchvision Dataset class wrapper. |
|
||||
| [Ptv Dataset Loader](ptv_dataset_loader.md) | A node to combine the dataset and data loader into a single node. |
|
||||
| [Ptv Image Folder Dataset](ptv_image_folder_dataset.md) | A Torchvision ImageFolder Dataset class wrapper. |
|
||||
| [Ptv Transforms To Tensor](ptv_transforms_to_tensor.md) | Transforms elements of dataset to PyTorch tensors. |
|
||||
## PyTorch wrapper - Transform
|
||||
| Node | Description |
|
||||
|
||||
@@ -0,0 +1,28 @@
|
||||
# Ptv Image Folder Dataset
|
||||
A Torchvision ImageFolder Dataset class wrapper.
|
||||
|
||||
Parameters
|
||||
|
||||
**root**
|
||||
Specify the root directory that contains subdirectories which of which contains images for each class.
|
||||
The subdirectory name needs to be named after each class.
|
||||
|
||||
**transform**
|
||||
Data transforms e.g. transform to PyTorch tensor, normalize image.
|
||||
Plug in the PTV Transforms node that contains Torchvision Transforms functionality.
|
||||
|
||||
## Input
|
||||
| Name | Data type |
|
||||
|---|---|
|
||||
| root | String |
|
||||
| transform | Ptvtransform |
|
||||
|
||||
## Output
|
||||
| Data type |
|
||||
|---|
|
||||
| Ptvdataset |
|
||||
|
||||
<HR>
|
||||
Category: PyTorch wrapper - Training
|
||||
|
||||
ComfyUI Data Analysis Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.
|
||||
Reference in New Issue
Block a user