48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
|
import numpy as np
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
|
||
|
|
def load_text_numpy(path, delimiter, dtype):
|
||
|
|
if isinstance(delimiter, (tuple, list)):
|
||
|
|
for d in delimiter:
|
||
|
|
try:
|
||
|
|
ground_truth_rect = np.loadtxt(path, delimiter=d, dtype=dtype)
|
||
|
|
return ground_truth_rect
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
raise Exception('Could not read file {}'.format(path))
|
||
|
|
else:
|
||
|
|
ground_truth_rect = np.loadtxt(path, delimiter=delimiter, dtype=dtype)
|
||
|
|
return ground_truth_rect
|
||
|
|
|
||
|
|
|
||
|
|
def load_text_pandas(path, delimiter, dtype):
|
||
|
|
if isinstance(delimiter, (tuple, list)):
|
||
|
|
for d in delimiter:
|
||
|
|
try:
|
||
|
|
ground_truth_rect = pd.read_csv(path, delimiter=d, header=None, dtype=dtype, na_filter=False,
|
||
|
|
low_memory=False).values
|
||
|
|
return ground_truth_rect
|
||
|
|
except Exception as e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
raise Exception('Could not read file {}'.format(path))
|
||
|
|
else:
|
||
|
|
ground_truth_rect = pd.read_csv(path, delimiter=delimiter, header=None, dtype=dtype, na_filter=False,
|
||
|
|
low_memory=False).values
|
||
|
|
return ground_truth_rect
|
||
|
|
|
||
|
|
|
||
|
|
def load_text(path, delimiter=' ', dtype=np.float32, backend='numpy'):
|
||
|
|
if backend == 'numpy':
|
||
|
|
return load_text_numpy(path, delimiter, dtype)
|
||
|
|
elif backend == 'pandas':
|
||
|
|
return load_text_pandas(path, delimiter, dtype)
|
||
|
|
|
||
|
|
|
||
|
|
def load_str(path):
|
||
|
|
with open(path, "r") as f:
|
||
|
|
text_str = f.readline().strip().lower()
|
||
|
|
return text_str
|