Merge pull request #1 from ncl935/dev

ctgName checking in CreateTensor.py
aquaskyline · Aug 1, 2018 · a713423 · a713423
2 parents 0ef3fed + 9d9eb57
commit a713423
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dataPrepScripts/CreateTensor.py b/dataPrepScripts/CreateTensor.py
@@ -64,6 +64,7 @@ def GetCandidate(args, beginToEnd):
         fo = sys.stdin
     for row in fo:
         row = row.split()
+        if args.ctgName != row[0]: continue
         pos = int(row[1])
         if args.ctgStart != None and pos < args.ctgStart: continue
         if args.ctgEnd != None and pos > args.ctgEnd: continue