from annotator.uniformer.mmcv.utils import collect_env as collect_base_env | |
from annotator.uniformer.mmcv.utils import get_git_hash | |
import annotator.uniformer.mmseg as mmseg | |
def collect_env(): | |
"""Collect the information of the running environments.""" | |
env_info = collect_base_env() | |
env_info['MMSegmentation'] = f'{mmseg.__version__}+{get_git_hash()[:7]}' | |
return env_info | |
if __name__ == '__main__': | |
for name, val in collect_env().items(): | |
print('{}: {}'.format(name, val)) | |