utils.py
443 Bytes
'''
_*_coding:utf-8 _*_
@Time :2022/1/28 19:58
@Author : qiaofengsheng
@File :utils.py
@Software :PyCharm
'''
import torch
import yaml
from PIL import Image
from torch.nn.functional import one_hot
def load_config_util(config_path):
config_file = open(config_path, 'r', encoding='utf-8')
config_data = yaml.load(config_file)
return config_data
def label_one_hot(label):
return one_hot(torch.tensor(label))