Spaces:
Runtime error
Runtime error
Nicolò
commited on
Commit
•
5505fbb
1
Parent(s):
5105c39
typo + PEP8
Browse files- gan_vs_real_detector.py +1 -4
gan_vs_real_detector.py
CHANGED
@@ -116,14 +116,11 @@ class Detector:
|
|
116 |
|
117 |
|
118 |
def main():
|
119 |
-
|
120 |
parser = argparse.ArgumentParser()
|
121 |
-
parser.add_argument('--img_path', help='
|
122 |
args = parser.parse_args()
|
123 |
|
124 |
img_path = args.img_path
|
125 |
-
# debug img_path on fermi:
|
126 |
-
# img_path = '/home/nbonettini/nvidia_temp/nvidia-alias-free-gan/faces/alias-free-r-afhqv2-512x512/seed40000.png'
|
127 |
|
128 |
detector = Detector()
|
129 |
score = detector.synth_real_detector(img_path)
|
|
|
116 |
|
117 |
|
118 |
def main():
|
|
|
119 |
parser = argparse.ArgumentParser()
|
120 |
+
parser.add_argument('--img_path', help='Path to the test image', required=True)
|
121 |
args = parser.parse_args()
|
122 |
|
123 |
img_path = args.img_path
|
|
|
|
|
124 |
|
125 |
detector = Detector()
|
126 |
score = detector.synth_real_detector(img_path)
|