Spaces:
Running
on
Zero
Running
on
Zero
Update src/pixel3dmm/network_inference.py
Browse files
src/pixel3dmm/network_inference.py
CHANGED
@@ -88,10 +88,6 @@ def normals_n_uvs(cfg, model):
|
|
88 |
image_names = os.listdir(f'{IMAGE_FOLDER}')
|
89 |
image_names.sort()
|
90 |
|
91 |
-
if os.path.exists(out_folders[prediction_type]):
|
92 |
-
if len(os.listdir(out_folders[prediction_type])) == len(image_names):
|
93 |
-
return
|
94 |
-
|
95 |
for i in tqdm(range(len(image_names))):
|
96 |
try:
|
97 |
img = np.array(Image.open(f'{IMAGE_FOLDER}/{image_names[i]}').resize((512, 512))) / 255 # need 512,512 images as input; normalize to [0, 1] range
|
|
|
88 |
image_names = os.listdir(f'{IMAGE_FOLDER}')
|
89 |
image_names.sort()
|
90 |
|
|
|
|
|
|
|
|
|
91 |
for i in tqdm(range(len(image_names))):
|
92 |
try:
|
93 |
img = np.array(Image.open(f'{IMAGE_FOLDER}/{image_names[i]}').resize((512, 512))) / 255 # need 512,512 images as input; normalize to [0, 1] range
|