MCPcopy Create free account

hub / github.com/S1mpleyang/ConstructionActionRecognition / functions

Functions90 in github.com/S1mpleyang/ConstructionActionRecognition

⨍ Functions 90◇ Types & classes 15
↓ 15 callersMethodupdate
(self, val, n=1)
utils.py:22
↓ 7 callersMethodwrite
(self, values)
utils.py:59
↓ 4 callersMethodload_all_frames
(self, video_path)
dataloaders/all_frame_dataset.py:109
↓ 3 callersMethodlog
(self, values)
utils.py:41
↓ 3 callersFunctiontsn_seg
按照 tsn 方法划分数据 :param seed: :param frame_len: 总的帧长度 :param num_frames: 选择的帧长度 :return:
Cluster/cluster.py:49
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
utils.py:124
↓ 2 callersFunctionavg_seg
均匀采样 :param frame_len: :param num_frames: :return:
Cluster/cluster.py:103
↓ 2 callersFunctionbuild_config
(dataset)
configuration.py:1
↓ 2 callersFunctionget_opt
()
setting.py:26
↓ 2 callersMethodload_all_frames
(self, video_path)
dataloaders/video_dataset.py:99
↓ 2 callersFunctionmax_pool_seg
Temporal Aggregate Representation 每一段的视频帧 池化为一帧 :param frame_len: :param num_frames: :return:
Cluster/cluster.py:88
↓ 2 callersFunctionrandom_seg
随机采样 :param frame_len: :param num_frames: :return:
Cluster/cluster.py:68
↓ 2 callersFunctionread
(path)
draw/dataset_analysis/num_of_clip.py:16
↓ 1 callersMethodbuild_all_clip
(self, video_path)
dataloaders/all_frame_dataset.py:176
↓ 1 callersMethodbuild_pool_clip
(self, video_path)
dataloaders/all_frame_dataset.py:169
↓ 1 callersMethodbuild_tsn_clip
(self, video_path)
dataloaders/video_dataset.py:133
↓ 1 callersMethodbuild_tsn_clip
(self, video_path)
dataloaders/all_frame_dataset.py:160
↓ 1 callersFunctioncalculate_precision_recall_f1
(output, targets, num_classes, threshold=0.5)
utils.py:86
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
dataloaders/video_dataset.py:57
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
dataloaders/all_frame_dataset.py:59
↓ 1 callersFunctioneach_accuracy
Computes the precision@k for the specified values of k
utils.py:141
↓ 1 callersMethodfreeze_backbone
冻结特征提取 :return:
Cluster/cluster.py:33
↓ 1 callersFunctiongenerate
(dataset, split)
dataloaders/generate_train_split.py:6
↓ 1 callersFunctionget_lr
(optimizer)
utils.py:108
↓ 1 callersFunctionget_mean_std
(value_scale, dataset)
setting.py:108
↓ 1 callersFunctionget_test_utils
(opt, cfg)
main.py:336
↓ 1 callersFunctionget_test_utils
(opt, cfg)
evaluate.py:92
↓ 1 callersFunctionget_train_utils
(opt, cfg, model_parameters)
main.py:267
↓ 1 callersFunctionget_val_utils
(opt, cfg)
main.py:318
↓ 1 callersFunctionmain_worker
:param opt: :return:
main.py:355
↓ 1 callersFunctionmain_worker
(opt)
evaluate.py:113
↓ 1 callersFunctionnormalize
(vid, mean, std)
dataloaders/video_dataset.py:41
↓ 1 callersFunctionnormalize
(vid, mean, std)
dataloaders/all_frame_dataset.py:43
↓ 1 callersFunctionparse_opts
()
opts.py:5
↓ 1 callersFunctionplot_confusion_matrix
- cm : 计算出的混淆矩阵的值 - classes : 混淆矩阵中每一行每一列对应的列 - normalize : True:显示百分比, False:显示个数
draw/draw_cm.py:5
↓ 1 callersFunctionplot_roc
( model_name, label, predict, color, linestyle=":", line
draw/roc_plot.py:22
↓ 1 callersMethodpool_seg
(self, frames, num_frames)
dataloaders/all_frame_dataset.py:145
↓ 1 callersFunctionread_log
读取日志文件,进行数据重组,写入mysql :return:
draw/loss_curve.py:7
↓ 1 callersMethodreset
(self)
utils.py:16
↓ 1 callersMethodreset
(self)
Cluster/cluster.py:166
↓ 1 callersFunctionresize
(frames, size, interpolation='bicubic')
dataloaders/video_dataset.py:15
↓ 1 callersFunctionresize
(frames, size, interpolation='bicubic')
dataloaders/all_frame_dataset.py:16
↓ 1 callersFunctionresume_model
(resume_path, model)
setting.py:82
↓ 1 callersFunctionresume_train_utils
(resume_path, begin_epoch, optimizer, scheduler)
setting.py:95
↓ 1 callersFunctionsave_checkpoint
(save_file_path, epoch, arch, model, optimizer, scheduler)
setting.py:12
↓ 1 callersFunctiontest_epoch
(epoch, data_loader, model, criterion, device,
main.py:165
↓ 1 callersFunctiontest_epoch
( data_loader, model, device, model_type, num_classes, l
evaluate.py:18
↓ 1 callersFunctionto_normalized_float_tensor
(vid)
dataloaders/video_dataset.py:32
↓ 1 callersFunctionto_normalized_float_tensor
(vid)
dataloaders/all_frame_dataset.py:33
↓ 1 callersFunctiontrain_epoch
(epoch, data_loader, model, criterion, optimiz
main.py:18
↓ 1 callersMethodtsn_seg
(self, frames, num_frames)
dataloaders/video_dataset.py:118
↓ 1 callersMethodtsn_seg
(self, frames, num_frames)
dataloaders/all_frame_dataset.py:127
↓ 1 callersFunctionval_epoch
(epoch, data_loader, model, criterion, device,
main.py:93
↓ 1 callersFunctionvideo_time
(file_path)
draw/dataset_analysis/num_of_clip.py:5
Method__call__
(self, vid)
dataloaders/video_dataset.py:28
Method__call__
(self, vid)
dataloaders/video_dataset.py:37
Method__call__
(self, vid)
dataloaders/video_dataset.py:53
Method__call__
(self, vid)
dataloaders/all_frame_dataset.py:29
Method__call__
(self, vid)
dataloaders/all_frame_dataset.py:39
Method__call__
(self, vid)
dataloaders/all_frame_dataset.py:55
Method__del
(self)
utils.py:38
Method__del
(self)
utils.py:56
Method__getitem__
(self, index)
dataloaders/video_dataset.py:146
Method__getitem__
:param index: :return: clips [bs, c, T, h, w] , label: np.zeros(num_classes)
dataloaders/all_frame_dataset.py:190
Method__init__
(self, num_classes=1000, at_type="DTM", lstm_channel=16)
STR_transformer.py:9
Method__init__
(self)
utils.py:13
Method__init__
(self, path, header)
utils.py:31
Method__init__
(self, path)
utils.py:53
Method__init__
(self, size)
dataloaders/video_dataset.py:25
Method__init__
(self, mean, std)
dataloaders/video_dataset.py:49
Method__init__
(self, cfg, data_split, data_percentage, num_frames, skip_frames, input_size, shuffle=False)
dataloaders/video_dataset.py:64
Method__init__
(self, size)
dataloaders/all_frame_dataset.py:26
Method__init__
(self, mean, std)
dataloaders/all_frame_dataset.py:51
Method__init__
(self, cfg, data_split, data_percentage, num_frames, seg_method, input_size, shuffle=False, seed=None)
dataloaders/all_frame_dataset.py:66
Method__init__
(self, backbone='resnet')
Cluster/cluster.py:13
Method__init__
(self)
Cluster/cluster.py:163
Method__len__
(self)
dataloaders/video_dataset.py:96
Method__len__
(self)
dataloaders/all_frame_dataset.py:106
Methodbuild_consecutive_clips
(self, video_path)
dataloaders/video_dataset.py:139
Methodbuild_consecutive_clips
(self, video_path)
dataloaders/all_frame_dataset.py:183
Functioncalculate_accuracy
(outputs, targets)
utils.py:64
Functioncalculate_precision_and_recall
(outputs, targets, pos_label=1)
utils.py:76
Methodforward
(self, x_raw)
STR_transformer.py:58
Methodforward
(self, x)
Cluster/cluster.py:41
Methodfreeze_model
(self, x)
STR_transformer.py:53
Functionjson_serial
(obj)
setting.py:8
Functionpartialclass
(cls, *args, **kwargs)
utils.py:117
Functionpool_seg
:param frames: tensor:Tensor :param num_frames: number:int :param seg_method: ["tsn", "avg", "random", "max_pool", "mean_pool", ]
Cluster/cluster.py:120
Methodupdate
(self, val, n=1)
Cluster/cluster.py:172
Functionworker_init_fn
(worker_id)
utils.py:98