diff --git a/__init__.py b/__init__.py index 2fb9ec7..3356d69 100644 --- a/__init__.py +++ b/__init__.py @@ -10,7 +10,7 @@ NODE_CLASS_MAPPINGS = { } NODE_DISPLAY_NAME_MAPPINGS = { "VFrame Loader With Mask Editor" : "SequentialImageLoader", - "Video Loader With Mask Editor" : "VideoExtractor", + "Video Loader With Mask Editor" : "VideoFrameExtractor", } __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] diff --git a/js/main.js b/js/main.js index 01e852b..7d83747 100644 --- a/js/main.js +++ b/js/main.js @@ -78,7 +78,7 @@ async function uplaodSequentialDataCore(node, inputName, inputData, app, fileInp document.body.append(fileInput); // Create the button widget for selecting the files - let uploadWidget = node.addWidget("button", "choose image directory to upload", "image", () => { + let uploadWidget = node.addWidget("button", fileInput['data-disp-msg'], "image", () => { fileInput.click(); }); uploadWidget.serialize = false; @@ -239,6 +239,7 @@ app.registerExtension({ type: "file", style: "display: none", webkitdirectory: true, + 'data-disp-msg': "choose image directory to upload", onchange: async () => { if (fileInput.files.length) { const vframeId = Date.now(); @@ -308,6 +309,7 @@ app.registerExtension({ type: "file", style: "display: none", accept: "video/*", + 'data-disp-msg': "choose mp4 to upload", onchange: async () => { if (fileInput.files.length) { const file = fileInput.files[0]; diff --git a/readme.md b/readme.md index 71684b8..a3989c7 100644 --- a/readme.md +++ b/readme.md @@ -9,6 +9,7 @@ Add this repository to the custom_nodes/ directory. ## Usage ### About "SequentialImageLoader" Node +This is in the Image category. #### input * sequence_id: Please ignore (used for internal processing only). @@ -21,6 +22,10 @@ Add this repository to the custom_nodes/ directory. * mask_images: Masks for each frame are output as images. You may need to convert them to mask data using a Mask To Image node, for example. * image_count: Number of processed frames. +### About "VideoFrameExtractor" Node +Instead of specifying a directory containing frames, you specify a video file. Currently, only mp4 format is supported. +Otherwise, it is the same as the SequentialImageLoader Node. + ### About the Mask Editor 1. After loading the frames, right-click the node and select "Open In MaskEditor". 2. In the editor that appears, perform masking and sketching as needed. It is based on the standard Mask Editor.