Files
ieemoo-ai-searchv2/cirtorch/utils/general.py
2022-11-22 15:32:06 +08:00

34 lines
894 B
Python
Executable File

import os
import hashlib
def get_root():
return os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))))
def get_data_root():
return os.path.join(get_root(), 'data')
def htime(c):
c = round(c)
days = c // 86400
hours = c // 3600 % 24
minutes = c // 60 % 60
seconds = c % 60
if days > 0:
return '{:d}d {:d}h {:d}m {:d}s'.format(days, hours, minutes, seconds)
if hours > 0:
return '{:d}h {:d}m {:d}s'.format(hours, minutes, seconds)
if minutes > 0:
return '{:d}m {:d}s'.format(minutes, seconds)
return '{:d}s'.format(seconds)
def sha256_hash(filename, block_size=65536, length=8):
sha256 = hashlib.sha256()
with open(filename, 'rb') as f:
for block in iter(lambda: f.read(block_size), b''):
sha256.update(block)
return sha256.hexdigest()[:length-1]