Directory fix

TextRearrange/DBSCAN.py

@@ -8,8 +8,8 @@ import seaborn as sns
 import matplotlib.pyplot  as plt
 from sklearn.cluster import DBSCAN
 from sklearn.preprocessing import StandardScaler
-import lib.arrange as DbscanArrange
-from lib import directories as Dir
+import arrange as DbscanArrange
+import directories as Dir
 
 """
 

app.py

@@ -10,7 +10,7 @@ def HCR(im):
     ps.textRearrange()
     ps.textRecognition()
     hcr = ps.getHcrResult(Dir.txt_file_path)
-    subprocess.call('cd'+ Dir.yolo_dir, shell=True)
+    #subprocess.call('cd'+ Dir.yolo_dir, shell=True)
     #clearDir()
 
     return hcr

directories.py

@@ -1,11 +1,11 @@
-home_dir = "/HCR"
-yolo_dir = "/HCR/TextDetection/"
-input_img = "/HCR/TextDetection/cookie/user_input.jpg"
+home_dir = "~"
+yolo_dir = "/TextDetection"
+input_img = "/TextDetection/cookie/user_input.jpg"
 detect_model_dir = "/runs/wordDetection/weights/best.pt"
 cropped_img_folder_name = "user_output"
 cropped_img_path = "/runs/detect/" + cropped_img_folder_name
 folder_path = cropped_img_path + "/crops/word"
-DBSCAN_dir = "/HCR/TextRearrange"
-recog_dir = "/HCR/TextRecognition"
-recog_model_dir='/HCR/TextRecognition/best_accuracy_s/best_accuracy_s.pth'
-txt_file_path = "/HCR/TextRecognition/log_demo_result.txt"  
+DBSCAN_dir = "/TextRearrange"
+recog_dir = "/TextRecognition"
+recog_model_dir = "/TextRecognition/best_accuracy_s/best_accuracy_s.pth"
+txt_file_path = "/TextRecognition/log_demo_result.txt"

process.py

@@ -20,7 +20,8 @@ def textDetection(im):
 
     #change dir to yolo folder
     #yolo_dir = "/HCR/TextDetection/"
-    subprocess.call('cd'+ Dir.yolo_dir, shell=True)
+    subprocess.call('pwd', shell=True)
+    subprocess.call('cd '+ Dir.yolo_dir, shell=True)
 
     #transfrom ndarray type to PIL type
     im = Image.fromarray(im)
@@ -30,7 +31,7 @@ def textDetection(im):
     im.save("user_input.jpg", 'JPEG')
 
     #yolo_dir = "/HCR/TextDetection/"
-    subprocess.call('cd'+ Dir.yolo_dir, shell=True)
+    subprocess.call('cd '+ Dir.yolo_dir, shell=True)
 
     # (Shell) run detect.py to get cropped word images
     subprocess.call(['python','detect.py',
@@ -52,14 +53,14 @@ def textDetection(im):
     #return Image.fromarray(results.imgs[0])
 
 def textRearrange():
-    subprocess.call('cd' + Dir.DBSCAN_dir, shell=True)
+    subprocess.call('cd ' + Dir.DBSCAN_dir, shell=True)
     subprocess.call(['python','DBSCAN.py'])
 
 def textRecognition():
     #%cd /content/drive/MyDrive/KITA/Text/lmdb/deep-text-recognition-benchmark
     subprocess.call('cd '+Dir.recog_dir, shell=True)
     #!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
-    subprocess.call('CUDA_VISIBLE_DEVICES="" python3 demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder ' + Dir.home_dir + Dir.cropped_img_path + '/crops/word --saved_model '+ Dir.recog_model_dir, shell=True)
+    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)
 
 def getHcrResult(file_path):#*#
     texts = ""