nyonyong commited on
Commit
1fcea5a
1 Parent(s): 6dc60e4

Directory fix

Browse files
Files changed (1) hide show
  1. process.py +11 -11
process.py CHANGED
@@ -20,22 +20,22 @@ def textDetection(im):
20
 
21
  #change dir to yolo folder
22
  #yolo_dir = "/HCR/TextDetection/"
23
- subprocess.call('pwd', shell=True)
24
- subprocess.call('ls', shell=True)
25
- subprocess.call('sudo cd '+ Dir.yolo_dir, shell=True)
26
 
27
  #transfrom ndarray type to PIL type
28
  im = Image.fromarray(im)
29
 
30
  # save input image to cookie folder
31
- subprocess.call('cd cookie', shell=True)
32
- im.save("user_input.jpg", 'JPEG')
33
 
34
  #yolo_dir = "/HCR/TextDetection/"
35
- subprocess.call('cd '+ Dir.yolo_dir, shell=True)
36
 
37
  # (Shell) run detect.py to get cropped word images
38
- subprocess.call(['python','detect.py',
39
  #User Input Data : /text_detection/cookie
40
  '--source','/cookie',
41
  #Text Detection Model : /runs/wordDetection/weights/best.pt
@@ -54,14 +54,14 @@ def textDetection(im):
54
  #return Image.fromarray(results.imgs[0])
55
 
56
  def textRearrange():
57
- subprocess.call('cd ' + Dir.DBSCAN_dir, shell=True)
58
- subprocess.call(['python','DBSCAN.py'])
59
 
60
  def textRecognition():
61
  #%cd /content/drive/MyDrive/KITA/Text/lmdb/deep-text-recognition-benchmark
62
- subprocess.call('cd '+Dir.recog_dir, shell=True)
63
  #!CUDA_VISIBLE_DEVICES=0 python3 demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder /content/drive/MyDrive/KITA/Text/YOLO/runs/detect/youtube_data2/crops/word --saved_model /content/drive/MyDrive/KITA/Text/best_accuracy_s/best_accuracy_s.pth
64
- subprocess.call('CUDA_VISIBLE_DEVICES="" python3 demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder ' + Dir.yolo_dir + Dir.cropped_img_path + '/crops/word --saved_model '+ Dir.recog_model_dir, shell=True)
65
 
66
  def getHcrResult(file_path):#*#
67
  texts = ""
 
20
 
21
  #change dir to yolo folder
22
  #yolo_dir = "/HCR/TextDetection/"
23
+ #subprocess.call('pwd', shell=True)
24
+ #subprocess.call('ls', shell=True)
25
+ ## subprocess.call('sudo cd '+ Dir.yolo_dir, shell=True)
26
 
27
  #transfrom ndarray type to PIL type
28
  im = Image.fromarray(im)
29
 
30
  # save input image to cookie folder
31
+ ## subprocess.call('cd cookie', shell=True)
32
+ im.save(Dir.yolo_dir+"/cookie/user_input.jpg", 'JPEG')
33
 
34
  #yolo_dir = "/HCR/TextDetection/"
35
+ ## subprocess.call('cd '+ Dir.yolo_dir, shell=True)
36
 
37
  # (Shell) run detect.py to get cropped word images
38
+ subprocess.call(['python', Dir.yolo_dir+'/detect.py',
39
  #User Input Data : /text_detection/cookie
40
  '--source','/cookie',
41
  #Text Detection Model : /runs/wordDetection/weights/best.pt
 
54
  #return Image.fromarray(results.imgs[0])
55
 
56
  def textRearrange():
57
+ ## subprocess.call('cd ' + Dir.DBSCAN_dir, shell=True)
58
+ subprocess.call(['python', Dir.DBSCAN_dir+'/DBSCAN.py'])
59
 
60
  def textRecognition():
61
  #%cd /content/drive/MyDrive/KITA/Text/lmdb/deep-text-recognition-benchmark
62
+ ## subprocess.call('cd '+Dir.recog_dir, shell=True)
63
  #!CUDA_VISIBLE_DEVICES=0 python3 demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder /content/drive/MyDrive/KITA/Text/YOLO/runs/detect/youtube_data2/crops/word --saved_model /content/drive/MyDrive/KITA/Text/best_accuracy_s/best_accuracy_s.pth
64
+ subprocess.call('CUDA_VISIBLE_DEVICES="" python3 '+ Dir.recog_dir +'/demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder ' + Dir.yolo_dir + Dir.cropped_img_path + '/crops/word --saved_model '+ Dir.recog_model_dir, shell=True)
65
 
66
  def getHcrResult(file_path):#*#
67
  texts = ""