Added Ptv Transforms Resize node to resize image data. Added workflows and PyTorch model training tutorials.
This commit is contained in:
@@ -243,6 +243,7 @@ from .modules.pytorch_wrapper.pto_adam import PtoAdam
|
||||
from .modules.pytorch_wrapper.ptv_dataset import PtvDataset
|
||||
from .modules.pytorch_wrapper.ptv_dataset_loader import PtvDatasetLoader
|
||||
from .modules.pytorch_wrapper.ptv_image_folder_dataset import PtvImageFolderDataset
|
||||
from .modules.pytorch_wrapper.ptv_transforms_resize import PtvTransformsResize
|
||||
from .modules.pytorch_wrapper.ptv_transforms_to_tensor import PtvTransformsToTensor
|
||||
from .modules.seaborn_wrapper.sns_bar import SNSBar
|
||||
from .modules.seaborn_wrapper.sns_boxplot import SNSBoxplot
|
||||
@@ -494,6 +495,7 @@ NODE_CLASS_MAPPINGS: Dict[str, Type[T]] = {
|
||||
"PtvDataset": PtvDataset,
|
||||
"PtvDatasetLoader": PtvDatasetLoader,
|
||||
"PtvImageFolderDataset": PtvImageFolderDataset,
|
||||
"PtvTransformsResize": PtvTransformsResize,
|
||||
"PtvTransformsToTensor": PtvTransformsToTensor,
|
||||
"PyDatetimeToString": PyDatetimeToString,
|
||||
"PyFloatToString": PyFloatToString,
|
||||
@@ -754,6 +756,7 @@ NODE_DISPLAY_NAME_MAPPINGS: Dict[str, str] = {
|
||||
"PtvDataset": "Ptv Dataset",
|
||||
"PtvDatasetLoader": "Ptv Dataset Loader",
|
||||
"PtvImageFolderDataset": "Ptv Image Folder Dataset",
|
||||
"PtvTransformsResize": "Ptv Transforms Resize",
|
||||
"PtvTransformsToTensor": "Ptv Transforms To Tensor",
|
||||
"PyDatetimeToString": "Py Datetime To String",
|
||||
"PyFloatToString": "Py Float To String",
|
||||
|
||||
Reference in New Issue
Block a user