Skip to content

Commit

Permalink
change ambigous name
Browse files Browse the repository at this point in the history
  • Loading branch information
@NevermindNilas
NevermindNilas committed Jun 3, 2024
1 parent 7116ad5 commit b951d59
Show file tree
Hide file tree
Showing 4 changed files with 38 additions and 38 deletions.
6 changes: 3 additions & 3 deletions PARAMETERS.MD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -64,10 +64,10 @@ The Anime Scripter is a powerful tool for enhancing and manipulating videos with
- `--segment` (action=store_true): Enable segmentation.
- `--segment_method` (str, default="anime", choices=["anime", "anime-tensorrt"])L Choose the desired segmentation method.

#### Scene Change Detection:
#### Auto Cut Clips Detection:

- `--scenechange` (action=store_true): Enable scene change detection.
- `--scenechange_sens` (float, default=40): Sensitivity of scene change detection.
- `--autoclip` (action=store_true): Enable scene change detection.
- `--autoclip_sens` (float, default=40): Sensitivity of scene change detection.

#### Depth Map Generation:

Expand Down
46 changes: 23 additions & 23 deletions TheAnimeScripter.jsx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ var TheAnimeScripter = (function () {
var outputFolder = app.settings.haveSetting(scriptName, "outputFolder") ? app.settings.getSetting(scriptName, "outputFolder") : "undefined";
var theAnimeScripterPath = app.settings.haveSetting(scriptName, "theAnimeScripterPath") ? app.settings.getSetting(scriptName, "theAnimeScripterPath") : "undefined";
var segmentValue = 0;
var sceneChangeValue = 0;
var autoClipValue = 0;
var depthValue = 0;
var customModelPath = "undefined";
var consentData = app.settings.haveSetting(scriptName, "consentData") ? app.settings.getSetting(scriptName, "consentData") : false;
Expand Down Expand Up @@ -126,13 +126,13 @@ var TheAnimeScripter = (function () {
buttonSegment.preferredSize.width = 105;
buttonSegment.alignment = ["center", "top"];

var buttonSceneChange = panelPostProcess.add("button", undefined, undefined, {
name: "buttonSceneChange"
var buttonAutoClip = panelPostProcess.add("button", undefined, undefined, {
name: "buttonAutoClip"
});

buttonSceneChange.text = "Auto Cut";
buttonSceneChange.preferredSize.width = 105;
buttonSceneChange.alignment = ["center", "top"];
buttonAutoClip.text = "Auto Cut";
buttonAutoClip.preferredSize.width = 105;
buttonAutoClip.alignment = ["center", "top"];

buttonGetVideo = panelPostProcess.add("button", undefined, undefined, {
name: "buttonGetVideo"
Expand Down Expand Up @@ -321,15 +321,15 @@ var TheAnimeScripter = (function () {
}
}

createSlider(generalPanel, "Sharpenening Sensitivity", "Sharpen", 50);
createSlider(generalPanel, "Sharpenening Sensitivity", "SharpenSens", 50);
createSlider(generalPanel, "Deduplication Sensitivity", "DedupSens", 50);
createSlider(generalPanel, "Auto Cut Sensitivity", "SceneChange", 50);
createSlider(generalPanel, "Auto Cut Sensitivity", "AutoClipSens", 50);

var sharpenSensValue = function () {
return labelValues["Sharpen"];
return labelValues["SharpenSens"];
};
var sceneChangeSensValue = function () {
return labelValues["SceneChange"];
return labelValues["AutoClipSens"];
};
var dedupSensValue = function () {
return labelValues["DedupSens"];
Expand Down Expand Up @@ -647,8 +647,8 @@ var TheAnimeScripter = (function () {
settingsWindow.show();
};

buttonSceneChange.onClick = function () {
sceneChangeValue = 1;
buttonAutoClip.onClick = function () {
autoClipValue = 1;
startChain();
}

Expand Down Expand Up @@ -840,7 +840,7 @@ var TheAnimeScripter = (function () {
"--sharpen_sens", sharpenSensValue(),
"--depth_method", depthModel().toLowerCase(),
"--encode_method", encoderMethod().toLowerCase(),
"--scenechange_sens", sceneChangeSensValue(),
"--autoclip_sens", sceneChangeSensValue(),
"--resize_method", resizeMethod().toLowerCase(),
"--resize_factor", resizeValue(),
"--nt", threadsValue(),
Expand Down Expand Up @@ -887,8 +887,8 @@ var TheAnimeScripter = (function () {
attempt.push("--depth");
}

if (sceneChangeValue == 1) {
attempt.push("--scenechange");
if (autoClipValue == 1) {
attempt.push("--autoclip");
}

if (customModelPath && customModelPath !== "undefined") {
Expand All @@ -906,24 +906,24 @@ var TheAnimeScripter = (function () {
}

while (true) {
if (sceneChangeValue == 1) {
var sceneChangeLogPath = theAnimeScripterPath + "\\scenechangeresults.txt";
var sceneChangeLog = new File(sceneChangeLogPath);
sceneChangeLog.open("r");
if (autoClipValue == 1) {
var autoClipLogPath = theAnimeScripterPath + "\\autoclipresults.txt";
var autoClipLog = new File(autoClipLogPath);
autoClipLog.open("r");

inPoint = layer.inPoint;
outPoint = layer.outPoint;

while (!sceneChangeLog.eof) {
var line = sceneChangeLog.readln();
while (!autoClipLog.eof) {
var line = autoClipLog.readln();
var timestamp = parseFloat(line) + inPoint;
var duplicateLayer = layer.duplicate();
layer.outPoint = timestamp;
duplicateLayer.inPoint = timestamp;

layer = duplicateLayer;
}
sceneChangeLog.close();
autoClipLog.close();
break;
} else {
var maxAttempts = 3;
Expand Down Expand Up @@ -967,7 +967,7 @@ var TheAnimeScripter = (function () {
}
i++;
}
sceneChangeValue = 0;
autoClipValue = 0;
segmentValue = 0;
depthValue = 0;
}
Expand Down
14 changes: 7 additions & 7 deletions main.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,8 +62,8 @@ def __init__(self, args):
self.sharpen = args.sharpen
self.sharpen_sens = args.sharpen_sens
self.segment = args.segment
self.scenechange = args.scenechange
self.scenechange_sens = args.scenechange_sens
self.autoclip = args.autoclip
self.autoclip_sens = args.autoclip_sens
self.depth = args.depth
self.depth_method = args.depth_method
self.encode_method = args.encode_method
Expand Down Expand Up @@ -105,13 +105,13 @@ def __init__(self, args):
)

if self.scenechange:
from src.scenechange.scene_change import Scenechange
from src.autoclip.autoclip import AutoClip

logging.info("Detecting scene changes")

Scenechange(
AutoClip(
self.input,
self.scenechange_sens,
self.autoclip_sens,
mainPath,
self.inpoint,
self.outpoint,
Expand Down Expand Up @@ -344,12 +344,12 @@ def start(self):
"--segment", action="store_true", help="Segment the video", required=False
)
argparser.add_argument(
"--scenechange",
"--autoclip",
action="store_true",
help="Detect scene changes",
required=False,
)
argparser.add_argument("--scenechange_sens", type=float, default=50)
argparser.add_argument("--autoclip_sens", type=float, default=50)
argparser.add_argument(
"--depth",
action="store_true",
Expand Down
10 changes: 5 additions & 5 deletions src/scenechange/scene_change.py → src/autoclip/autoclip.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,17 +4,17 @@
from scenedetect.detectors import ContentDetector


class Scenechange:
def __init__(self, input, scenechange_sens, output_dir, inPoint, outPoint):
class AutoClip:
def __init__(self, input, autoclip_sens, output_dir, inPoint, outPoint):
self.input = input
self.scenechange_sens = scenechange_sens
self.autoclip_sens = autoclip_sens
self.output_dir = output_dir
self.inPoint = inPoint
self.outPoint = outPoint

video_manager = VideoManager([self.input])
scene_manager = SceneManager()
scene_manager.add_detector(ContentDetector(threshold=self.scenechange_sens))
scene_manager.add_detector(ContentDetector(threshold=self.autoclip_sens))

if self.outPoint != 0:
start_time = FrameTimecode(self.inPoint, video_manager.get_framerate())
Expand All @@ -29,7 +29,7 @@ def __init__(self, input, scenechange_sens, output_dir, inPoint, outPoint):

scene_list = scene_manager.get_scene_list()

with open(os.path.join(self.output_dir, "scenechangeresults.txt"), "w") as f:
with open(os.path.join(self.output_dir, "autoclipresults.txt"), "w") as f:
for i, scene in enumerate(scene_list):
start_time = scene[0].get_seconds()
end_time = scene[1].get_seconds()
Expand Down

0 comments on commit b951d59

Please sign in to comment.