Marthee commited on
Commit
d28f107
·
verified ·
1 Parent(s): 493e2a6

Update doors_fasterrcnn.py

Browse files
Files changed (1) hide show
  1. doors_fasterrcnn.py +6 -1
doors_fasterrcnn.py CHANGED
@@ -218,7 +218,12 @@ def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpa
218
 
219
  df_doors = df_doors.fillna(' ')
220
  gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(df_doors , pdf_name,pdfpath)
221
- return annotatedimg, pdf_document , spreadsheet_url
 
 
 
 
 
222
 
223
  # model_path = '/content/drive/MyDrive/combined.pth'
224
  # #pdf_name = data
 
218
 
219
  df_doors = df_doors.fillna(' ')
220
  gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(df_doors , pdf_name,pdfpath)
221
+ list1=pd.DataFrame(columns=['content', 'id', 'subject'])
222
+ for page in pdf_document:
223
+ for annot in page.annots():
224
+ list1.loc[len(list1)] =annot.info
225
+
226
+ return annotatedimg, pdf_document , spreadsheet_url, list1
227
 
228
  # model_path = '/content/drive/MyDrive/combined.pth'
229
  # #pdf_name = data