提交 15be3c54 编写于 作者: qq_25193841's avatar qq_25193841

Auto stash before merge of "table" and "origin/new"

上级 e15cf0d1
...@@ -2131,7 +2131,7 @@ class MainWindow(QMainWindow): ...@@ -2131,7 +2131,7 @@ class MainWindow(QMainWindow):
TableRec_excel_dir = self.lastOpenDir + '/tableRec_excel_output/' TableRec_excel_dir = self.lastOpenDir + '/tableRec_excel_output/'
os.makedirs(TableRec_excel_dir, exist_ok=True) os.makedirs(TableRec_excel_dir, exist_ok=True)
filename = os.path.basename(self.filePath) filename, _ = os.path.splitext(os.path.basename(self.filePath))
excel_path = TableRec_excel_dir + '{}.xlsx'.format(filename) excel_path = TableRec_excel_dir + '{}.xlsx'.format(filename)
if res is None: if res is None:
...@@ -2203,19 +2203,26 @@ class MainWindow(QMainWindow): ...@@ -2203,19 +2203,26 @@ class MainWindow(QMainWindow):
return return
# automatically open excel annotation file # automatically open excel annotation file
try: if platform.system() == 'Windows':
import win32com.client try:
except: import win32com.client
print("CANNOT OPEN .xlsx. It could be one of the following reasons: " \ except:
"Only support Windows | No python win32com") print("CANNOT OPEN .xlsx. It could be one of the following reasons: " \
"Only support Windows | No python win32com")
try: try:
xl = win32com.client.Dispatch("Excel.Application") xl = win32com.client.Dispatch("Excel.Application")
xl.Visible = True xl.Visible = True
xl.Workbooks.Open(excel_path) xl.Workbooks.Open(excel_path)
except: # excelEx = "You need to show the excel executable at this point"
print("CANNOT OPEN .xlsx. It could be the following reasons: " \ # subprocess.Popen([excelEx, excel_path])
".xlsx is not existed")
# os.startfile(excel_path)
except:
print("CANNOT OPEN .xlsx. It could be the following reasons: " \
".xlsx is not existed")
else:
os.system('open ' + os.path.normpath(excel_path))
print('time cost: ', time.time() - start) print('time cost: ', time.time() - start)
...@@ -2337,7 +2344,7 @@ class MainWindow(QMainWindow): ...@@ -2337,7 +2344,7 @@ class MainWindow(QMainWindow):
imgid = 0 imgid = 0
for image_path in labeldict.keys(): for image_path in labeldict.keys():
# load csv annotations # load csv annotations
filename = os.path.basename(image_path) filename, _ = os.path.splitext(os.path.basename(image_path))
csv_path = os.path.join(TableRec_excel_dir, filename + '.xlsx') csv_path = os.path.join(TableRec_excel_dir, filename + '.xlsx')
if not os.path.exists(csv_path): if not os.path.exists(csv_path):
msg = 'ERROR, Can not find ' + csv_path msg = 'ERROR, Can not find ' + csv_path
...@@ -2347,7 +2354,7 @@ class MainWindow(QMainWindow): ...@@ -2347,7 +2354,7 @@ class MainWindow(QMainWindow):
# read xlsx file, convert to HTML # read xlsx file, convert to HTML
xd = pd.ExcelFile(csv_path) xd = pd.ExcelFile(csv_path)
df = xd.parse() df = xd.parse()
structure = df.to_html() structure = df.to_html(index = False)
# load box annotations # load box annotations
cells = [] cells = []
...@@ -2371,7 +2378,7 @@ class MainWindow(QMainWindow): ...@@ -2371,7 +2378,7 @@ class MainWindow(QMainWindow):
imgid += 1 imgid += 1
# save json # save json
with open("{}/annotation.json".format(self.lastOpenDir), "w") as fid: with open("{}/annotation.json".format(self.lastOpenDir), "w", encoding='utf-8') as fid:
fid.write(json.dumps(json_results)) fid.write(json.dumps(json_results))
msg = 'JSON sucessfully saved in {}/annotation.json'.format(self.lastOpenDir) msg = 'JSON sucessfully saved in {}/annotation.json'.format(self.lastOpenDir)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册