From 385effc52674bce14f58fd9860f24dc7e72e4210 Mon Sep 17 00:00:00 2001 From: FuouM Date: Sat, 10 Aug 2024 19:38:18 +0700 Subject: [PATCH] rename modules folder --- .gitignore | 4 ++-- constants.py | 8 ++++---- inference_articulate.py | 8 ++++---- inference_spline.py | 8 ++++---- {articulate_module => module_articulate}/avd_network.py | 0 {articulate_module => module_articulate}/batchnorm.py | 0 .../bg_motion_predictor.py | 0 {articulate_module => module_articulate}/comm.py | 0 .../config/taichi256.yaml | 0 .../config/ted-youtube384.yaml | 0 .../config/ted384.yaml | 0 .../config/vox256.yaml | 0 {articulate_module => module_articulate}/generator.py | 0 {articulate_module => module_articulate}/models/.gitkeep | 0 .../pixelwise_flow_predictor.py | 0 .../region_predictor.py | 0 {articulate_module => module_articulate}/util.py | 0 {articulate_module => module_articulate}/visualizer.py | 0 {spline_module => module_spline}/avd_network.py | 0 {spline_module => module_spline}/bg_motion_predictor.py | 0 {spline_module => module_spline}/config/mgif-256.yaml | 0 {spline_module => module_spline}/config/taichi-256.yaml | 0 {spline_module => module_spline}/config/ted-384.yaml | 0 {spline_module => module_spline}/config/vox-256.yaml | 0 {spline_module => module_spline}/dense_motion.py | 0 {spline_module => module_spline}/inpainting_network.py | 0 {spline_module => module_spline}/keypoint_detector.py | 0 {spline_module => module_spline}/model.py | 0 {spline_module => module_spline}/models/.gitkeep | 0 {spline_module => module_spline}/util.py | 0 30 files changed, 14 insertions(+), 14 deletions(-) rename {articulate_module => module_articulate}/avd_network.py (100%) rename {articulate_module => module_articulate}/batchnorm.py (100%) rename {articulate_module => module_articulate}/bg_motion_predictor.py (100%) rename {articulate_module => module_articulate}/comm.py (100%) rename {articulate_module => module_articulate}/config/taichi256.yaml (100%) rename {articulate_module => module_articulate}/config/ted-youtube384.yaml (100%) rename {articulate_module => module_articulate}/config/ted384.yaml (100%) rename {articulate_module => module_articulate}/config/vox256.yaml (100%) rename {articulate_module => module_articulate}/generator.py (100%) rename {articulate_module => module_articulate}/models/.gitkeep (100%) rename {articulate_module => module_articulate}/pixelwise_flow_predictor.py (100%) rename {articulate_module => module_articulate}/region_predictor.py (100%) rename {articulate_module => module_articulate}/util.py (100%) rename {articulate_module => module_articulate}/visualizer.py (100%) rename {spline_module => module_spline}/avd_network.py (100%) rename {spline_module => module_spline}/bg_motion_predictor.py (100%) rename {spline_module => module_spline}/config/mgif-256.yaml (100%) rename {spline_module => module_spline}/config/taichi-256.yaml (100%) rename {spline_module => module_spline}/config/ted-384.yaml (100%) rename {spline_module => module_spline}/config/vox-256.yaml (100%) rename {spline_module => module_spline}/dense_motion.py (100%) rename {spline_module => module_spline}/inpainting_network.py (100%) rename {spline_module => module_spline}/keypoint_detector.py (100%) rename {spline_module => module_spline}/model.py (100%) rename {spline_module => module_spline}/models/.gitkeep (100%) rename {spline_module => module_spline}/util.py (100%) diff --git a/.gitignore b/.gitignore index 5c409c1..1a2d0fd 100644 --- a/.gitignore +++ b/.gitignore @@ -1,7 +1,7 @@ # Stuff -articulate_module/models/*.pth -spline_module/models/*.tar +module_articulate/models/*.pth +module_spline/models/*.tar checkpoints/* !checkpoints/place_checkpoints_here.txt diff --git a/constants.py b/constants.py index 9876432..93b184b 100644 --- a/constants.py +++ b/constants.py @@ -99,11 +99,11 @@ seg_model_dict = { "vox-adv": seg10_segment_names, } -ARTICULATE_MODEL_PATH = "articulate_module/models/vox256.pth" -ARTICULATE_CFG_PATH = "articulate_module/config/vox256.yaml" +ARTICULATE_MODEL_PATH = "module_articulate/models/vox256.pth" +ARTICULATE_CFG_PATH = "module_articulate/config/vox256.yaml" SPLINE_MODES = ['relative', 'standard', 'avd'] SPLINE_DEFAULT = 'relative' -SPLINE_MODEL_PATH = "spline_module/models/vox.pth.tar" -SPLINE_CFG_PATH = "spline_module/config/vox-256.yaml" \ No newline at end of file +SPLINE_MODEL_PATH = "module_spline/models/vox.pth.tar" +SPLINE_CFG_PATH = "module_spline/config/vox-256.yaml" \ No newline at end of file diff --git a/inference_articulate.py b/inference_articulate.py index 0d2c1a4..c357546 100644 --- a/inference_articulate.py +++ b/inference_articulate.py @@ -5,10 +5,10 @@ import yaml from comfy.utils import ProgressBar from scipy.spatial import ConvexHull -from .articulate_module.avd_network import AVDNetwork -from .articulate_module.bg_motion_predictor import BGMotionPredictor -from .articulate_module.generator import Generator -from .articulate_module.region_predictor import RegionPredictor +from .module_articulate.avd_network import AVDNetwork +from .module_articulate.bg_motion_predictor import BGMotionPredictor +from .module_articulate.generator import Generator +from .module_articulate.region_predictor import RegionPredictor from .sync_batchnorm.replicate import DataParallelWithCallback diff --git a/inference_spline.py b/inference_spline.py index 80bec5f..d2595e9 100644 --- a/inference_spline.py +++ b/inference_spline.py @@ -7,10 +7,10 @@ from comfy.utils import ProgressBar from .inference_fomm import find_best_frame -from .spline_module.avd_network import AVDNetwork -from .spline_module.dense_motion import DenseMotionNetwork -from .spline_module.keypoint_detector import KPDetector -from .spline_module.inpainting_network import InpaintingNetwork +from .module_spline.avd_network import AVDNetwork +from .module_spline.dense_motion import DenseMotionNetwork +from .module_spline.keypoint_detector import KPDetector +from .module_spline.inpainting_network import InpaintingNetwork def spline_inference( diff --git a/articulate_module/avd_network.py b/module_articulate/avd_network.py similarity index 100% rename from articulate_module/avd_network.py rename to module_articulate/avd_network.py diff --git a/articulate_module/batchnorm.py b/module_articulate/batchnorm.py similarity index 100% rename from articulate_module/batchnorm.py rename to module_articulate/batchnorm.py diff --git a/articulate_module/bg_motion_predictor.py b/module_articulate/bg_motion_predictor.py similarity index 100% rename from articulate_module/bg_motion_predictor.py rename to module_articulate/bg_motion_predictor.py diff --git a/articulate_module/comm.py b/module_articulate/comm.py similarity index 100% rename from articulate_module/comm.py rename to module_articulate/comm.py diff --git a/articulate_module/config/taichi256.yaml b/module_articulate/config/taichi256.yaml similarity index 100% rename from articulate_module/config/taichi256.yaml rename to module_articulate/config/taichi256.yaml diff --git a/articulate_module/config/ted-youtube384.yaml b/module_articulate/config/ted-youtube384.yaml similarity index 100% rename from articulate_module/config/ted-youtube384.yaml rename to module_articulate/config/ted-youtube384.yaml diff --git a/articulate_module/config/ted384.yaml b/module_articulate/config/ted384.yaml similarity index 100% rename from articulate_module/config/ted384.yaml rename to module_articulate/config/ted384.yaml diff --git a/articulate_module/config/vox256.yaml b/module_articulate/config/vox256.yaml similarity index 100% rename from articulate_module/config/vox256.yaml rename to module_articulate/config/vox256.yaml diff --git a/articulate_module/generator.py b/module_articulate/generator.py similarity index 100% rename from articulate_module/generator.py rename to module_articulate/generator.py diff --git a/articulate_module/models/.gitkeep b/module_articulate/models/.gitkeep similarity index 100% rename from articulate_module/models/.gitkeep rename to module_articulate/models/.gitkeep diff --git a/articulate_module/pixelwise_flow_predictor.py b/module_articulate/pixelwise_flow_predictor.py similarity index 100% rename from articulate_module/pixelwise_flow_predictor.py rename to module_articulate/pixelwise_flow_predictor.py diff --git a/articulate_module/region_predictor.py b/module_articulate/region_predictor.py similarity index 100% rename from articulate_module/region_predictor.py rename to module_articulate/region_predictor.py diff --git a/articulate_module/util.py b/module_articulate/util.py similarity index 100% rename from articulate_module/util.py rename to module_articulate/util.py diff --git a/articulate_module/visualizer.py b/module_articulate/visualizer.py similarity index 100% rename from articulate_module/visualizer.py rename to module_articulate/visualizer.py diff --git a/spline_module/avd_network.py b/module_spline/avd_network.py similarity index 100% rename from spline_module/avd_network.py rename to module_spline/avd_network.py diff --git a/spline_module/bg_motion_predictor.py b/module_spline/bg_motion_predictor.py similarity index 100% rename from spline_module/bg_motion_predictor.py rename to module_spline/bg_motion_predictor.py diff --git a/spline_module/config/mgif-256.yaml b/module_spline/config/mgif-256.yaml similarity index 100% rename from spline_module/config/mgif-256.yaml rename to module_spline/config/mgif-256.yaml diff --git a/spline_module/config/taichi-256.yaml b/module_spline/config/taichi-256.yaml similarity index 100% rename from spline_module/config/taichi-256.yaml rename to module_spline/config/taichi-256.yaml diff --git a/spline_module/config/ted-384.yaml b/module_spline/config/ted-384.yaml similarity index 100% rename from spline_module/config/ted-384.yaml rename to module_spline/config/ted-384.yaml diff --git a/spline_module/config/vox-256.yaml b/module_spline/config/vox-256.yaml similarity index 100% rename from spline_module/config/vox-256.yaml rename to module_spline/config/vox-256.yaml diff --git a/spline_module/dense_motion.py b/module_spline/dense_motion.py similarity index 100% rename from spline_module/dense_motion.py rename to module_spline/dense_motion.py diff --git a/spline_module/inpainting_network.py b/module_spline/inpainting_network.py similarity index 100% rename from spline_module/inpainting_network.py rename to module_spline/inpainting_network.py diff --git a/spline_module/keypoint_detector.py b/module_spline/keypoint_detector.py similarity index 100% rename from spline_module/keypoint_detector.py rename to module_spline/keypoint_detector.py diff --git a/spline_module/model.py b/module_spline/model.py similarity index 100% rename from spline_module/model.py rename to module_spline/model.py diff --git a/spline_module/models/.gitkeep b/module_spline/models/.gitkeep similarity index 100% rename from spline_module/models/.gitkeep rename to module_spline/models/.gitkeep diff --git a/spline_module/util.py b/module_spline/util.py similarity index 100% rename from spline_module/util.py rename to module_spline/util.py