alexnasa commited on
Commit
14320b5
·
verified ·
1 Parent(s): cca6d90

Update src/pixel3dmm/network_inference.py

Browse files
Files changed (1) hide show
  1. src/pixel3dmm/network_inference.py +0 -4
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