Functions — github.com/TingsongYu/PyTorch_Tutorial

CodeHub MCPcopy Create free account

hub / github.com/TingsongYu/PyTorch_Tutorial / functions

Functions40 in github.com/TingsongYu/PyTorch_Tutorial

⨍ Functions 40 ◇ Types & classes 9

↓ 4 callersFunctionshow_confMat

(confusion_mat, classes, set_name, out_dir)

Code/utils/utils.py:112

↓ 4 callersFunctionvalidate

对一批数据进行预测，返回混淆矩阵以及Accuracy :param net: :param data_loader: :param set_name: eg: 'valid' 'train' 'tesst :param classes_name:

Code/utils/utils.py:73

↓ 2 callersFunctiongen_txt

(txt_path, img_dir)

Code/1_data_prepare/1_3_generate_txt.py:14

↓ 2 callersFunctionmy_mkdir

Code/1_data_prepare/1_1_cifar10_to_png.py:25

↓ 2 callersFunctionunpickle

Code/1_data_prepare/1_1_cifar10_to_png.py:20

↓ 1 callersFunctioncomp_class_vec

计算类向量 :param ouput_vec: tensor :param index: int，指定类别 :return: tensor

Code/4_viewer/6_hook_for_grad_cam.py:88

↓ 1 callersFunctiongen_cam

依据梯度和特征图，生成cam :param feature_map: np.array， in [C, H, W] :param grads: np.array， in [C, H, W] :return: np.array, [H, W]

Code/4_viewer/6_hook_for_grad_cam.py:108

↓ 1 callersFunctionimg_preprocess

读取图片，转为模型可读的形式 :param img_in: ndarray, [H, W, C] :return: PIL.image

Code/4_viewer/6_hook_for_grad_cam.py:49

↓ 1 callersFunctionimg_transform

将img进行预处理，并转换成模型输入所需的形式—— B*C*H*W :param img_roi: np.array :return:

Code/4_viewer/6_hook_for_grad_cam.py:36

↓ 1 callersMethodinitialize_weights

Code/main_training/main.py:90

↓ 1 callersMethodinitialize_weights

Code/4_viewer/2_visual_weights.py:31

↓ 1 callersFunctionmakedir

Code/1_data_prepare/1_2_split_dataset.py:21

↓ 1 callersFunctionnormalize_invert

(tensor, mean, std)

Code/utils/utils.py:141

↓ 1 callersFunctionshow_cam_on_image

(img, mask, out_dir)

Code/4_viewer/6_hook_for_grad_cam.py:74

↓ 1 callersMethodzero_param

Code/3_optimizer/3_2_optimizer/4_load_state_dict.py:23

Method__getitem__

Code/utils/utils.py:60

Method__getitem__

Code/1_data_prepare/1_3_mydataset.py:19

Code/utils/utils.py:14

(self, txt_path, transform = None, target_transform = None)

Code/utils/utils.py:48

Code/3_optimizer/3_2_optimizer/3_state_dict.py:9

Code/3_optimizer/3_2_optimizer/4_load_state_dict.py:11

(self, txt_path, transform=None, target_transform=None)

Code/1_data_prepare/1_3_mydataset.py:7

Code/main_training/main.py:70

Code/4_viewer/6_hook_for_grad_cam.py:16

Code/4_viewer/2_visual_weights.py:11

Code/2_model/2_finetune.py:67

Code/utils/utils.py:69

Code/1_data_prepare/1_3_mydataset.py:28

Functionbackward_hook

(module, grad_in, grad_out)

Code/4_viewer/6_hook_for_grad_cam.py:66

Functionfarward_hook

(module, input, output)

Code/4_viewer/6_hook_for_grad_cam.py:70

Code/utils/utils.py:24

Code/3_optimizer/3_2_optimizer/3_state_dict.py:15

Code/3_optimizer/3_2_optimizer/4_load_state_dict.py:17

Code/main_training/main.py:80

Code/4_viewer/6_hook_for_grad_cam.py:26

Code/4_viewer/2_visual_weights.py:21

Code/2_model/2_finetune.py:77

Methodinitialize_weights

Code/utils/utils.py:34

Methodinitialize_weights

Code/2_model/2_finetune.py:87

Functionshow_confMat

可视化混淆矩阵，保存png格式 :param confusion_mat: nd-array :param classes_name: list,各类别名称 :param set_name: str, eg: 'valid', 'train' :param

Code/4_viewer/5_Show_ConfMat.py:8