mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
disable abbreviating csv inference output with ellipses
This commit is contained in:
parent
ff77dfa825
commit
8ce197e33a
@ -12,6 +12,7 @@ import os
|
||||
import time
|
||||
from contextlib import suppress
|
||||
from functools import partial
|
||||
from sys import maxsize
|
||||
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
@ -343,6 +344,7 @@ def main():
|
||||
|
||||
|
||||
def save_results(df, results_filename, results_format='csv', filename_col='filename'):
|
||||
np.set_printoptions(threshold=maxsize)
|
||||
results_filename += _FMT_EXT[results_format]
|
||||
if results_format == 'parquet':
|
||||
df.set_index(filename_col).to_parquet(results_filename)
|
||||
|
Loading…
x
Reference in New Issue
Block a user