mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

disable abbreviating csv inference output with ellipses

Browse Source
This commit is contained in:
Josua Rieder 2025-01-14 17:39:48 +01:00
parent ff77dfa825
commit 8ce197e33a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -12,6 +12,7 @@ import os
import time import time
from contextlib import suppress from contextlib import suppress
from functools import partial from functools import partial
from sys import maxsize
import numpy as np import numpy as np
import pandas as pd import pandas as pd
@ -343,6 +344,7 @@ def main():
def save_results(df, results_filename, results_format='csv', filename_col='filename'): def save_results(df, results_filename, results_format='csv', filename_col='filename'):
np.set_printoptions(threshold=maxsize)
results_filename += _FMT_EXT[results_format] results_filename += _FMT_EXT[results_format]
if results_format == 'parquet': if results_format == 'parquet':
df.set_index(filename_col).to_parquet(results_filename) df.set_index(filename_col).to_parquet(results_filename)