Spaces:
Paused
Paused
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
|