Commit d6550b10 authored by Bucky Lee's avatar Bucky Lee
Browse files

feat: release v0.1.0

parent 265d2d71
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -17,11 +17,11 @@ def process(file_path, model_name):
    image.save('data/dataset/target/image1.png')
    preprocess('data/dataset/target/image1.png', 'data/dataset/target_mask/image1.png')
    predict('RCDG_model')
    img = Image.open('core/results/RCDG_drive/test_latest/images/image1_fake_TB.png')
    img = Image.open('results/RCDG_drive/test_latest/images/image1_fake_TB.png')
    img = img = img.resize((512, 512))
    img.save(str(f'data/processed/{file_name}.png'))
    os.remove('data/dataset/target/image1.png')
    os.remove('data/dataset/target_mask/image1.png')
    # os.remove('data/dataset/target/image1.png')
    # os.remove('data/dataset/target_mask/image1.png')

    return 'Success'

+0 −1
Original line number Diff line number Diff line
@@ -4,7 +4,6 @@ import cv2
import numpy as np
from scipy import ndimage


def preprocess(image_path, save_path):
    """

−59.3 KiB
Loading image diff...
−89.6 KiB
Loading image diff...
−90.4 KiB
Loading image diff...
Loading