Fix video background removal to use only allowed RGBA values
- Remove Transparent option [0,0,0,0] which is not allowed by API - Update background color options to show RGBA values in dropdown names - Change default background color from Transparent to White - Update fallback RGBA value to White [255,255,255,0]
This commit is contained in:
Sirshendu Ganguly
+12
-13
@@ -8,17 +8,16 @@ class videoBgRemoval:
|
||||
}
|
||||
|
||||
BACKGROUND_COLORS = {
|
||||
"Transparent": [0, 0, 0, 0],
|
||||
"Black": [0, 0, 0, 1],
|
||||
"White": [255, 255, 255, 0],
|
||||
"Gray": [128, 128, 128, 0],
|
||||
"Red": [255, 0, 0, 0],
|
||||
"Green": [0, 255, 0, 0],
|
||||
"Blue": [0, 0, 255, 0],
|
||||
"Yellow": [255, 255, 0, 0],
|
||||
"Cyan": [0, 255, 255, 0],
|
||||
"Magenta": [255, 0, 255, 0],
|
||||
"Orange": [255, 165, 0, 0],
|
||||
"Black [0, 0, 0, 1]": [0, 0, 0, 1],
|
||||
"White [255, 255, 255, 0]": [255, 255, 255, 0],
|
||||
"Gray [128, 128, 128, 0]": [128, 128, 128, 0],
|
||||
"Red [255, 0, 0, 0]": [255, 0, 0, 0],
|
||||
"Green [0, 255, 0, 0]": [0, 255, 0, 0],
|
||||
"Blue [0, 0, 255, 0]": [0, 0, 255, 0],
|
||||
"Yellow [255, 255, 0, 0]": [255, 255, 0, 0],
|
||||
"Cyan [0, 255, 255, 0]": [0, 255, 255, 0],
|
||||
"Magenta [255, 0, 255, 0]": [255, 0, 255, 0],
|
||||
"Orange [255, 165, 0, 0]": [255, 165, 0, 0],
|
||||
}
|
||||
|
||||
|
||||
@@ -36,7 +35,7 @@ class videoBgRemoval:
|
||||
},
|
||||
"optional": {
|
||||
"Background Color": (list(cls.BACKGROUND_COLORS.keys()), {
|
||||
"default": "Transparent",
|
||||
"default": "White",
|
||||
"tooltip": "Background color preset to use. Hex values are not supported.",
|
||||
}),
|
||||
"Output Format": (["mp4", "webm", "mov"], {
|
||||
@@ -66,7 +65,7 @@ class videoBgRemoval:
|
||||
modelAIR = self.RUNWARE_VRMBG_MODELS.get(modelName, "bria:51@1")
|
||||
|
||||
# Get background color RGBA
|
||||
rgba_color = self.BACKGROUND_COLORS.get(backgroundColor, [0, 0, 0, 0])
|
||||
rgba_color = self.BACKGROUND_COLORS.get(backgroundColor, [255, 255, 255, 0])
|
||||
|
||||
# Build the API request
|
||||
genConfig = [
|
||||
|
||||
@@ -1 +1 @@
|
||||
{"id":"a468db07-531a-4f8b-8513-003107cbfa22","revision":0,"last_node_id":5,"last_link_id":3,"nodes":[{"id":3,"type":"Runware Media Upload","pos":[92.27204895019531,542.3822021484375],"size":[270,58],"flags":{},"order":1,"mode":0,"inputs":[{"localized_name":"media","name":"media","shape":7,"type":"VIDEO,AUDIO","link":1},{"localized_name":"mediaUUID","name":"mediaUUID","shape":7,"type":"STRING","widget":{"name":"mediaUUID"},"link":null}],"outputs":[{"localized_name":"mediaUUID","name":"mediaUUID","type":"STRING","links":[2]}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"1c7da18f11d51514140944e0f9f90e62ae99572c","Node name for S&R":"Runware Media Upload"},"widgets_values":["b42cc934-5f2a-4b4f-b33d-d838433ce22e"],"bgcolor":"#5345bf"},{"id":4,"type":"LoadVideo","pos":[-223.58924865722656,531.5103759765625],"size":[274.080078125,442.0882263183594],"flags":{},"order":0,"mode":0,"inputs":[{"localized_name":"file","name":"file","type":"COMBO","widget":{"name":"file"},"link":null},{"localized_name":"choose file to upload","name":"upload","type":"IMAGEUPLOAD","widget":{"name":"upload"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[1]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"LoadVideo"},"widgets_values":["backgroundRemoval.mp4","image"]},{"id":5,"type":"SaveVideo","pos":[443.1638488769531,554.9398803710938],"size":[270,460.60870361328125],"flags":{},"order":3,"mode":0,"inputs":[{"localized_name":"video","name":"video","type":"VIDEO","link":3},{"localized_name":"filename_prefix","name":"filename_prefix","type":"STRING","widget":{"name":"filename_prefix"},"link":null},{"localized_name":"format","name":"format","type":"COMBO","widget":{"name":"format"},"link":null},{"localized_name":"codec","name":"codec","type":"COMBO","widget":{"name":"codec"},"link":null}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"SaveVideo"},"widgets_values":["video/ComfyUI","auto","auto"]},{"id":2,"type":"Runware Video Background Removal","pos":[74.94189453125,660.0944213867188],"size":[333.1080017089844,130],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"Video","name":"Video","type":"STRING","widget":{"name":"Video"},"link":2},{"localized_name":"Model","name":"Model","type":"COMBO","widget":{"name":"Model"},"link":null},{"localized_name":"Background Color","name":"Background Color","shape":7,"type":"COMBO","widget":{"name":"Background Color"},"link":null},{"localized_name":"Output Format","name":"Output Format","shape":7,"type":"COMBO","widget":{"name":"Output Format"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[3]}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"1c7da18f11d51514140944e0f9f90e62ae99572c","Node name for S&R":"Runware Video Background Removal"},"widgets_values":["","bria:51@1","Magenta","mp4"],"bgcolor":"#5345bf"}],"links":[[1,4,0,3,0,"VIDEO,AUDIO"],[2,3,0,2,0,"STRING"],[3,2,0,5,0,"VIDEO"]],"groups":[],"config":{},"extra":{"ds":{"scale":1.02884607845781,"offset":[636.1646218051795,-324.5220431776123]}},"version":0.4}
|
||||
{"id":"a468db07-531a-4f8b-8513-003107cbfa22","revision":0,"last_node_id":5,"last_link_id":3,"nodes":[{"id":3,"type":"Runware Media Upload","pos":[92.27204895019531,542.3822021484375],"size":[270,58],"flags":{},"order":1,"mode":0,"inputs":[{"localized_name":"media","name":"media","shape":7,"type":"VIDEO,AUDIO","link":1},{"localized_name":"mediaUUID","name":"mediaUUID","shape":7,"type":"STRING","widget":{"name":"mediaUUID"},"link":null}],"outputs":[{"localized_name":"mediaUUID","name":"mediaUUID","type":"STRING","links":[2]}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"1c7da18f11d51514140944e0f9f90e62ae99572c","Node name for S&R":"Runware Media Upload"},"widgets_values":["b42cc934-5f2a-4b4f-b33d-d838433ce22e"],"bgcolor":"#5345bf"},{"id":4,"type":"LoadVideo","pos":[-223.58924865722656,531.5103759765625],"size":[274.080078125,442.0882263183594],"flags":{},"order":0,"mode":0,"inputs":[{"localized_name":"file","name":"file","type":"COMBO","widget":{"name":"file"},"link":null},{"localized_name":"choose file to upload","name":"upload","type":"IMAGEUPLOAD","widget":{"name":"upload"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[1]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"LoadVideo"},"widgets_values":["backgroundRemoval.mp4","image"]},{"id":5,"type":"SaveVideo","pos":[443.1638488769531,554.9398803710938],"size":[270,460.60870361328125],"flags":{},"order":3,"mode":0,"inputs":[{"localized_name":"video","name":"video","type":"VIDEO","link":3},{"localized_name":"filename_prefix","name":"filename_prefix","type":"STRING","widget":{"name":"filename_prefix"},"link":null},{"localized_name":"format","name":"format","type":"COMBO","widget":{"name":"format"},"link":null},{"localized_name":"codec","name":"codec","type":"COMBO","widget":{"name":"codec"},"link":null}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"SaveVideo"},"widgets_values":["video/ComfyUI","auto","auto"]},{"id":2,"type":"Runware Video Background Removal","pos":[74.94189453125,660.0944213867188],"size":[333.1080017089844,130],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"Video","name":"Video","type":"STRING","widget":{"name":"Video"},"link":2},{"localized_name":"Model","name":"Model","type":"COMBO","widget":{"name":"Model"},"link":null},{"localized_name":"Background Color","name":"Background Color","shape":7,"type":"COMBO","widget":{"name":"Background Color"},"link":null},{"localized_name":"Output Format","name":"Output Format","shape":7,"type":"COMBO","widget":{"name":"Output Format"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[3]}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"1c7da18f11d51514140944e0f9f90e62ae99572c","Node name for S&R":"Runware Video Background Removal"},"widgets_values":["","bria:51@1","Orange [255, 165, 0, 0]","mp4"],"bgcolor":"#5345bf"}],"links":[[1,4,0,3,0,"VIDEO,AUDIO"],[2,3,0,2,0,"STRING"],[3,2,0,5,0,"VIDEO"]],"groups":[],"config":{},"extra":{"ds":{"scale":1.02884607845781,"offset":[440.951990678612,-413.0503791037476]}},"version":0.4}
|
||||
Reference in New Issue
Block a user