aryrk
commited on
Commit
·
c9b5e9c
1
Parent(s):
4c9b339
[debug]
Browse files
app.py
CHANGED
@@ -32,6 +32,7 @@ def reflection_removal(input_image):
|
|
32 |
shutil.copy(input_image, file_path)
|
33 |
|
34 |
input_filename = os.path.splitext(os.path.basename(file_path))[0]
|
|
|
35 |
|
36 |
cmd = [
|
37 |
"python", "test.py",
|
@@ -47,7 +48,11 @@ def reflection_removal(input_image):
|
|
47 |
|
48 |
for root, _, files in os.walk(RESULTS_DIR):
|
49 |
for file in files:
|
50 |
-
|
|
|
|
|
|
|
|
|
51 |
result_path = os.path.join(root, file)
|
52 |
return [Image.open(result_path)]
|
53 |
|
|
|
32 |
shutil.copy(input_image, file_path)
|
33 |
|
34 |
input_filename = os.path.splitext(os.path.basename(file_path))[0]
|
35 |
+
print(f"Processing {input_filename}...")
|
36 |
|
37 |
cmd = [
|
38 |
"python", "test.py",
|
|
|
48 |
|
49 |
for root, _, files in os.walk(RESULTS_DIR):
|
50 |
for file in files:
|
51 |
+
print(os.path.join(root, file))
|
52 |
+
|
53 |
+
for root, _, files in os.walk(RESULTS_DIR):
|
54 |
+
for file in files:
|
55 |
+
if file.startswith(input_filename) and file.endswith("_fake.png"):
|
56 |
result_path = os.path.join(root, file)
|
57 |
return [Image.open(result_path)]
|
58 |
|