From 6a2db5e1b78012b88bd583419a1e068d022db808 Mon Sep 17 00:00:00 2001
From: Vimal Manohar <vimal.manohar91@gmail.com>
Date: Fri, 21 Feb 2014 16:41:43 +0000
Subject: [PATCH] trunk/egs/babel: Minor change in segmentation.py to take
 python location from the environment

git-svn-id: https://svn.code.sf.net/p/kaldi/code/trunk@3565 5e6a8d80-dfce-4ca6-a32a-6e07a63d50c8
---
 egs/babel/s5b/local/resegment/segmentation.py | 58 ++++++++++---------
 1 file changed, 32 insertions(+), 26 deletions(-)

diff --git a/egs/babel/s5b/local/resegment/segmentation.py b/egs/babel/s5b/local/resegment/segmentation.py
index 440d33063..e40692f3b 100755
--- a/egs/babel/s5b/local/resegment/segmentation.py
+++ b/egs/babel/s5b/local/resegment/segmentation.py
@@ -1,4 +1,4 @@
-#! /usr/bin/python
+#! /usr/bin/env python
 
 # Copyright 2014  Vimal Manohar
 # Apache 2.0
@@ -6,7 +6,11 @@
 import os, glob, argparse, sys, re, time
 from argparse import ArgumentParser
 
-import numpy as np
+use_numpy = True
+try:
+  import numpy as np
+except ImportError:
+  use_numpy = False
 
 # Global stats for analysis taking RTTM file as reference
 global_analysis_get_initial_segments = None
@@ -157,18 +161,19 @@ class Analysis:
         max_length    = max([0]+self.type_counts[j][i])
         min_length    = min([10000]+self.type_counts[j][i])
         mean_length   = mean(self.type_counts[j][i])
-        try:
-          percentile25  = np.percentile(self.type_counts[j][i], 25)
-        except ValueError:
-          percentile25 = 0
-        try:
-          percentile50  = np.percentile(self.type_counts[j][i], 50)
-        except ValueError:
-          percentile50 = 0
-        try:
-          percentile75  = np.percentile(self.type_counts[j][i], 75)
-        except ValueError:
-          percentile75 = 0
+        if use_numpy:
+          try:
+            percentile25  = np.percentile(self.type_counts[j][i], 25)
+          except ValueError:
+            percentile25 = 0
+          try:
+            percentile50  = np.percentile(self.type_counts[j][i], 50)
+          except ValueError:
+            percentile50 = 0
+          try:
+            percentile75  = np.percentile(self.type_counts[j][i], 75)
+          except ValueError:
+            percentile75 = 0
 
         file_handle.write("File %s: %s : TypeStats: Type %d %d: Min: %4d Max: %4d Mean: %4d percentile25: %4d percentile50: %4d percentile75: %4d\n" % (self.file_id, self.prefix, j, i,  min_length, max_length, mean_length, percentile25, percentile50, percentile75))
       # End for loop over 9 different kinds of segments
@@ -183,18 +188,19 @@ class Analysis:
       self.max_length[i]    = max([0]+self.state_count[i])
       self.min_length[i]    = min([10000]+self.state_count[i])
       self.mean_length[i]   = mean(self.state_count[i])
-      try:
-        self.percentile25[i]  = np.percentile(self.state_count[i], 25)
-      except ValueError:
-        self.percentile25[i] = 0
-      try:
-        self.percentile50[i]  = np.percentile(self.state_count[i], 50)
-      except ValueError:
-        self.percentile50[i] = 0
-      try:
-        self.percentile75[i]  = np.percentile(self.state_count[i], 75)
-      except ValueError:
-        self.percentile75[i] = 0
+      if use_numpy:
+        try:
+          self.percentile25[i]  = np.percentile(self.state_count[i], 25)
+        except ValueError:
+          self.percentile25[i] = 0
+        try:
+          self.percentile50[i]  = np.percentile(self.state_count[i], 50)
+        except ValueError:
+          self.percentile50[i] = 0
+        try:
+          self.percentile75[i]  = np.percentile(self.state_count[i], 75)
+        except ValueError:
+          self.percentile75[i] = 0
 
       file_handle.write("File %s: %s : Length: Type %d: Min: %4d Max: %4d Mean: %4d percentile25: %4d percentile50: %4d percentile75: %4d\n" % (self.file_id, self.prefix, i,  self.min_length[i], self.max_length[i], self.mean_length[i], self.percentile25[i], self.percentile50[i], self.percentile75[i]))
     # End for loop over 9 cells