diff --git a/fsgan/criterions/__init__.py b/criterions/__init__.py similarity index 100% rename from fsgan/criterions/__init__.py rename to criterions/__init__.py diff --git a/fsgan/criterions/gan_loss.py b/criterions/gan_loss.py similarity index 100% rename from fsgan/criterions/gan_loss.py rename to criterions/gan_loss.py diff --git a/fsgan/criterions/vgg_loss.py b/criterions/vgg_loss.py similarity index 100% rename from fsgan/criterions/vgg_loss.py rename to criterions/vgg_loss.py diff --git a/fsgan/datasets/__init__.py b/datasets/__init__.py similarity index 100% rename from fsgan/datasets/__init__.py rename to datasets/__init__.py diff --git a/fsgan/datasets/appearance_map.py b/datasets/appearance_map.py similarity index 100% rename from fsgan/datasets/appearance_map.py rename to datasets/appearance_map.py diff --git a/fsgan/datasets/image_list_dataset.py b/datasets/image_list_dataset.py similarity index 100% rename from fsgan/datasets/image_list_dataset.py rename to datasets/image_list_dataset.py diff --git a/fsgan/datasets/image_seg_dataset.py b/datasets/image_seg_dataset.py similarity index 100% rename from fsgan/datasets/image_seg_dataset.py rename to datasets/image_seg_dataset.py diff --git a/fsgan/datasets/img_landmarks_transforms.py b/datasets/img_landmarks_transforms.py similarity index 100% rename from fsgan/datasets/img_landmarks_transforms.py rename to datasets/img_landmarks_transforms.py diff --git a/fsgan/datasets/img_lms_pose_transforms.py b/datasets/img_lms_pose_transforms.py similarity index 100% rename from fsgan/datasets/img_lms_pose_transforms.py rename to datasets/img_lms_pose_transforms.py diff --git a/fsgan/datasets/opencv_video_seq_dataset.py b/datasets/opencv_video_seq_dataset.py similarity index 100% rename from fsgan/datasets/opencv_video_seq_dataset.py rename to datasets/opencv_video_seq_dataset.py diff --git a/fsgan/datasets/seq_dataset.py b/datasets/seq_dataset.py similarity index 100% rename from fsgan/datasets/seq_dataset.py rename to datasets/seq_dataset.py diff --git a/fsgan/datasets/video_inference_dataset.py b/datasets/video_inference_dataset.py similarity index 100% rename from fsgan/datasets/video_inference_dataset.py rename to datasets/video_inference_dataset.py diff --git a/fsgan/experiments/reenactment/ijbc_msrunet_reenactment_attr.py b/experiments/reenactment/ijbc_msrunet_reenactment_attr.py similarity index 100% rename from fsgan/experiments/reenactment/ijbc_msrunet_reenactment_attr.py rename to experiments/reenactment/ijbc_msrunet_reenactment_attr.py diff --git a/fsgan/experiments/reenactment/ijbc_msrunet_reenactment_attr_no_seg.py b/experiments/reenactment/ijbc_msrunet_reenactment_attr_no_seg.py similarity index 100% rename from fsgan/experiments/reenactment/ijbc_msrunet_reenactment_attr_no_seg.py rename to experiments/reenactment/ijbc_msrunet_reenactment_attr_no_seg.py diff --git a/fsgan/experiments/reenactment/nfv_msrunet_reenactment_attr_no_seg_v2.1.py b/experiments/reenactment/nfv_msrunet_reenactment_attr_no_seg_v2.1.py similarity index 100% rename from fsgan/experiments/reenactment/nfv_msrunet_reenactment_attr_no_seg_v2.1.py rename to experiments/reenactment/nfv_msrunet_reenactment_attr_no_seg_v2.1.py diff --git a/fsgan/experiments/segmentation/celeba_unet.py b/experiments/segmentation/celeba_unet.py similarity index 100% rename from fsgan/experiments/segmentation/celeba_unet.py rename to experiments/segmentation/celeba_unet.py diff --git a/fsgan/experiments/swapping/ijbc_msrunet_blending.py b/experiments/swapping/ijbc_msrunet_blending.py similarity index 100% rename from fsgan/experiments/swapping/ijbc_msrunet_blending.py rename to experiments/swapping/ijbc_msrunet_blending.py diff --git a/fsgan/experiments/swapping/ijbc_msrunet_inpainting.py b/experiments/swapping/ijbc_msrunet_inpainting.py similarity index 100% rename from fsgan/experiments/swapping/ijbc_msrunet_inpainting.py rename to experiments/swapping/ijbc_msrunet_inpainting.py diff --git a/fsgan/inference/face_swapping.ipynb b/inference/face_swapping.ipynb similarity index 100% rename from fsgan/inference/face_swapping.ipynb rename to inference/face_swapping.ipynb diff --git a/fsgan/inference/reenact.py b/inference/reenact.py similarity index 100% rename from fsgan/inference/reenact.py rename to inference/reenact.py diff --git a/fsgan/inference/swap.py b/inference/swap.py similarity index 100% rename from fsgan/inference/swap.py rename to inference/swap.py diff --git a/fsgan/models/__init__.py b/models/__init__.py similarity index 100% rename from fsgan/models/__init__.py rename to models/__init__.py diff --git a/fsgan/models/classifier1d.py b/models/classifier1d.py similarity index 100% rename from fsgan/models/classifier1d.py rename to models/classifier1d.py diff --git a/fsgan/models/discriminators_pix2pix.py b/models/discriminators_pix2pix.py similarity index 100% rename from fsgan/models/discriminators_pix2pix.py rename to models/discriminators_pix2pix.py diff --git a/fsgan/models/hopenet.py b/models/hopenet.py similarity index 100% rename from fsgan/models/hopenet.py rename to models/hopenet.py diff --git a/fsgan/models/hrnet.py b/models/hrnet.py similarity index 100% rename from fsgan/models/hrnet.py rename to models/hrnet.py diff --git a/fsgan/models/msba.py b/models/msba.py similarity index 100% rename from fsgan/models/msba.py rename to models/msba.py diff --git a/fsgan/models/res_unet.py b/models/res_unet.py similarity index 100% rename from fsgan/models/res_unet.py rename to models/res_unet.py diff --git a/fsgan/models/res_unet_msba.py b/models/res_unet_msba.py similarity index 100% rename from fsgan/models/res_unet_msba.py rename to models/res_unet_msba.py diff --git a/fsgan/models/res_unet_split.py b/models/res_unet_split.py similarity index 100% rename from fsgan/models/res_unet_split.py rename to models/res_unet_split.py diff --git a/fsgan/models/simple_unet.py b/models/simple_unet.py similarity index 100% rename from fsgan/models/simple_unet.py rename to models/simple_unet.py diff --git a/fsgan/models/simple_unet_02.py b/models/simple_unet_02.py similarity index 100% rename from fsgan/models/simple_unet_02.py rename to models/simple_unet_02.py diff --git a/fsgan/models/vgg.py b/models/vgg.py similarity index 100% rename from fsgan/models/vgg.py rename to models/vgg.py diff --git a/fsgan/preprocess/__init__.py b/preprocess/__init__.py similarity index 100% rename from fsgan/preprocess/__init__.py rename to preprocess/__init__.py diff --git a/fsgan/preprocess/clear_cache.py b/preprocess/clear_cache.py similarity index 100% rename from fsgan/preprocess/clear_cache.py rename to preprocess/clear_cache.py diff --git a/fsgan/preprocess/crop_image_sequences.py b/preprocess/crop_image_sequences.py similarity index 100% rename from fsgan/preprocess/crop_image_sequences.py rename to preprocess/crop_image_sequences.py diff --git a/fsgan/preprocess/crop_video_sequences.py b/preprocess/crop_video_sequences.py similarity index 100% rename from fsgan/preprocess/crop_video_sequences.py rename to preprocess/crop_video_sequences.py diff --git a/fsgan/preprocess/crop_video_sequences_batch.py b/preprocess/crop_video_sequences_batch.py similarity index 100% rename from fsgan/preprocess/crop_video_sequences_batch.py rename to preprocess/crop_video_sequences_batch.py diff --git a/fsgan/preprocess/detections2sequences_1euro.py b/preprocess/detections2sequences_1euro.py similarity index 100% rename from fsgan/preprocess/detections2sequences_1euro.py rename to preprocess/detections2sequences_1euro.py diff --git a/fsgan/preprocess/detections2sequences_center.py b/preprocess/detections2sequences_center.py similarity index 100% rename from fsgan/preprocess/detections2sequences_center.py rename to preprocess/detections2sequences_center.py diff --git a/fsgan/preprocess/euler_sequences.py b/preprocess/euler_sequences.py similarity index 100% rename from fsgan/preprocess/euler_sequences.py rename to preprocess/euler_sequences.py diff --git a/fsgan/preprocess/preprocess_video.py b/preprocess/preprocess_video.py similarity index 99% rename from fsgan/preprocess/preprocess_video.py rename to preprocess/preprocess_video.py index d92ed4c..25aaa00 100644 --- a/fsgan/preprocess/preprocess_video.py +++ b/preprocess/preprocess_video.py @@ -18,7 +18,7 @@ import torch.nn as nn from torch.utils.data import DataLoader import torchvision.transforms as transforms -from face_detection_dsfd.face_detector import FaceDetector +from face_detection_dsfd.face_detection_dsfd.face_detector import FaceDetector from fsgan.utils.utils import set_device, load_model from fsgan.preprocess.detections2sequences_center import main as detections2sequences_main from fsgan.preprocess.crop_video_sequences import main as crop_video_sequences_main diff --git a/fsgan/preprocess/produce_train_val.py b/preprocess/produce_train_val.py similarity index 100% rename from fsgan/preprocess/produce_train_val.py rename to preprocess/produce_train_val.py diff --git a/fsgan/preprocess/render_sequences.py b/preprocess/render_sequences.py similarity index 100% rename from fsgan/preprocess/render_sequences.py rename to preprocess/render_sequences.py diff --git a/fsgan/preprocess/sequence_stats.py b/preprocess/sequence_stats.py similarity index 100% rename from fsgan/preprocess/sequence_stats.py rename to preprocess/sequence_stats.py diff --git a/fsgan/utils/__init__.py b/utils/__init__.py similarity index 100% rename from fsgan/utils/__init__.py rename to utils/__init__.py diff --git a/fsgan/utils/batch.py b/utils/batch.py similarity index 100% rename from fsgan/utils/batch.py rename to utils/batch.py diff --git a/fsgan/utils/bbox_utils.py b/utils/bbox_utils.py similarity index 100% rename from fsgan/utils/bbox_utils.py rename to utils/bbox_utils.py diff --git a/fsgan/utils/blur.py b/utils/blur.py similarity index 100% rename from fsgan/utils/blur.py rename to utils/blur.py diff --git a/fsgan/utils/confusionmatrix.py b/utils/confusionmatrix.py similarity index 100% rename from fsgan/utils/confusionmatrix.py rename to utils/confusionmatrix.py diff --git a/fsgan/utils/img_utils.py b/utils/img_utils.py similarity index 100% rename from fsgan/utils/img_utils.py rename to utils/img_utils.py diff --git a/fsgan/utils/iou_metric.py b/utils/iou_metric.py similarity index 100% rename from fsgan/utils/iou_metric.py rename to utils/iou_metric.py diff --git a/fsgan/utils/landmarks_utils.py b/utils/landmarks_utils.py similarity index 100% rename from fsgan/utils/landmarks_utils.py rename to utils/landmarks_utils.py diff --git a/fsgan/utils/obj_factory.py b/utils/obj_factory.py similarity index 100% rename from fsgan/utils/obj_factory.py rename to utils/obj_factory.py diff --git a/fsgan/utils/one_euro_filter.py b/utils/one_euro_filter.py similarity index 100% rename from fsgan/utils/one_euro_filter.py rename to utils/one_euro_filter.py diff --git a/fsgan/utils/seg_utils.py b/utils/seg_utils.py similarity index 100% rename from fsgan/utils/seg_utils.py rename to utils/seg_utils.py diff --git a/fsgan/utils/set_checkpoint_arch.py b/utils/set_checkpoint_arch.py similarity index 100% rename from fsgan/utils/set_checkpoint_arch.py rename to utils/set_checkpoint_arch.py diff --git a/fsgan/utils/temporal_smoothing.py b/utils/temporal_smoothing.py similarity index 100% rename from fsgan/utils/temporal_smoothing.py rename to utils/temporal_smoothing.py diff --git a/fsgan/utils/tensorboard_logger.py b/utils/tensorboard_logger.py similarity index 100% rename from fsgan/utils/tensorboard_logger.py rename to utils/tensorboard_logger.py diff --git a/fsgan/utils/utils.py b/utils/utils.py similarity index 100% rename from fsgan/utils/utils.py rename to utils/utils.py diff --git a/fsgan/utils/video_renderer.py b/utils/video_renderer.py similarity index 100% rename from fsgan/utils/video_renderer.py rename to utils/video_renderer.py diff --git a/fsgan/utils/video_utils.py b/utils/video_utils.py similarity index 100% rename from fsgan/utils/video_utils.py rename to utils/video_utils.py