Parcourir la source

bug fix

master
Chris il y a 8 ans
Parent
révision
8f810faaab
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  1. +1
    -0
      test_pretrained.py

+ 1
- 0
test_pretrained.py Voir le fichier

@@ -63,6 +63,7 @@ def main(args):
cv2.imwrite(out_fn, raw)

post, coords = post_process(raw)
print coords

out_fn = os.path.join(args.out_dir, f.replace('/','_')[:-4] + "_post.png")
cv2.imwrite(out_fn, post)


Chargement…
Annuler
Enregistrer