Chris 8 лет назад
Родитель
Сommit
8f810faaab
1 измененных файлов: 1 добавлений и 0 удалений
  1. +1
    -0
      test_pretrained.py

+ 1
- 0
test_pretrained.py Просмотреть файл

@@ -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)


Загрузка…
Отмена
Сохранить