Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dangweili/pedestrian-attribute-recognition-pytorch
/ functions
Functions
77 in github.com/dangweili/pedestrian-attribute-recognition-pytorch
⨍
Functions
77
◇
Types & classes
14
↓ 5 callers
Function
remove_fc
Remove the fc layer parameter from state_dict.
baseline/model/resnet.py:149
↓ 4 callers
Method
_make_layer
(self, block, planes, blocks, stride=1)
baseline/model/resnet.py:119
↓ 3 callers
Function
time_str
(fmt=None)
baseline/utils/utils.py:11
↓ 2 callers
Method
__init__
(self, block, layers, last_conv_stride=2)
baseline/model/resnet.py:98
↓ 2 callers
Function
attribute_evaluate_subfunc
evaluate the attribute recognition precision
script/experiment/train_deepmar_resnet50.py:301
↓ 2 callers
Method
close
(self)
baseline/utils/utils.py:163
↓ 2 callers
Function
conv3x3
3x3 convolution with padding
baseline/model/resnet.py:19
↓ 2 callers
Function
set_devices
Args: sys_device_ids: a tuple; which GPUs to use e.g. sys_device_ids = (), only use cpu sys_device_ids = (3,),
baseline/utils/utils.py:174
↓ 2 callers
Function
set_seed
(rand_seed)
baseline/utils/utils.py:32
↓ 2 callers
Function
transfer_optim_state
(state, device_id=-1)
baseline/utils/utils.py:198
↓ 1 callers
Function
adjust_lr_staircase
Multiplied by a factor at the beging of specified epochs. Different params groups specify thier own base learning rates. Args: pa
baseline/utils/utils.py:286
↓ 1 callers
Function
attribute_evaluate
(feat_func, dataset, **kwargs)
baseline/utils/evaluate.py:34
↓ 1 callers
Function
attribute_evaluate_lidw
Input: gt_result, pt_result, N*L, with 0/1 Output: result a dictionary, including label-based and instance-based evaluation
baseline/utils/evaluate.py:49
↓ 1 callers
Function
create_trainvaltest_split
create a dataset split file, which consists of index of the train/val/test splits
script/dataset/transform_rap.py:39
↓ 1 callers
Function
create_trainvaltest_split
create a dataset split file, which consists of index of the train/val/test splits
script/dataset/transform_pa100k.py:47
↓ 1 callers
Function
create_trainvaltest_split
create a dataset split file, which consists of index of the train/val/test splits
script/dataset/transform_rap2.py:39
↓ 1 callers
Function
create_trainvaltest_split
create a dataset split file, which consists of index of the train/val/test splits
script/dataset/transform_peta.py:38
↓ 1 callers
Function
extract_feat
extract feature for images
baseline/utils/evaluate.py:9
↓ 1 callers
Function
find_index
(seq, item)
baseline/utils/utils.py:168
↓ 1 callers
Function
generate_data_description
create a dataset description file, which consists of images, labels
script/dataset/transform_rap.py:16
↓ 1 callers
Function
generate_data_description
create a dataset description file, which consists of images, labels
script/dataset/transform_pa100k.py:16
↓ 1 callers
Function
generate_data_description
create a dataset description file, which consists of images, labels
script/dataset/transform_rap2.py:16
↓ 1 callers
Function
generate_data_description
create a dataset description file, which consists of images, labels
script/dataset/transform_peta.py:16
↓ 1 callers
Function
is_iterable
(obj)
baseline/utils/utils.py:19
↓ 1 callers
Function
load_ckpt
load state_dict of module & optimizer from file Args: modules_optims: A two-element list which contains module and optimizer
baseline/utils/utils.py:248
↓ 1 callers
Function
may_mkdir
(fname)
baseline/utils/utils.py:39
↓ 1 callers
Function
may_set_mode
maybe_modules, an object or a list of objects.
baseline/utils/utils.py:316
↓ 1 callers
Function
resnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
baseline/model/resnet.py:181
↓ 1 callers
Function
save_ckpt
save state_dict of modules/optimizers to file Args: modules_optims: a two-element list which contains a module and a optimizer
baseline/utils/utils.py:265
↓ 1 callers
Function
to_scalar
transform a 1-length pytorch Variable or Tensor to scalar
baseline/utils/utils.py:22
↓ 1 callers
Method
update
(self, val, n=1)
baseline/utils/utils.py:57
Method
__call__
(self, img)
baseline/dataset/add_transforms.py:12
Method
__call__
Args: img: a 3-dimensional torch tensor with shape [R,G,B]*H*W Returns: img: a 3-dimensional padded tensor wi
baseline/dataset/add_transforms.py:47
Method
__call__
(self, imgs)
baseline/model/DeepMAR.py:61
Method
__del__
(self)
baseline/utils/utils.py:134
Method
__enter__
(self)
baseline/utils/utils.py:137
Method
__exit__
(self, **args)
baseline/utils/utils.py:140
Method
__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is the index of the target class
baseline/dataset/Dataset.py:52
Method
__init__
(self)
script/experiment/train_deepmar_resnet50.py:38
Method
__init__
(self)
script/experiment/demo.py:25
Method
__init__
(self)
baseline/utils/utils.py:47
Method
__init__
(self, hist=0.99)
baseline/utils/utils.py:67
Method
__init__
(self, hist_size=100)
baseline/utils/utils.py:87
Method
__init__
(self, fpath=None, console='stdout', immediately_visiable=False)
baseline/utils/utils.py:117
Method
__init__
(self, size)
baseline/dataset/add_transforms.py:7
Method
__init__
(self, padding, fill=0)
baseline/dataset/add_transforms.py:21
Method
__init__
( self, dataset, partition, split='train', partition_idx=0, t
baseline/dataset/Dataset.py:12
Method
__init__
( self, **kwargs )
baseline/model/DeepMAR.py:11
Method
__init__
(self, model, **kwargs)
baseline/model/DeepMAR.py:58
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
baseline/model/resnet.py:28
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
baseline/model/resnet.py:60
Method
__len__
(self)
baseline/dataset/Dataset.py:76
Method
__repr__
(self)
baseline/dataset/add_transforms.py:10
Method
__repr__
(self)
baseline/dataset/add_transforms.py:44
Method
avg
(self)
baseline/utils/utils.py:102
Method
flush
(self)
baseline/utils/utils.py:156
Method
forward
(self, x)
baseline/model/DeepMAR.py:45
Method
forward
(self, x)
baseline/model/resnet.py:38
Method
forward
(self, x)
baseline/model/resnet.py:73
Method
forward
(self, x)
baseline/model/resnet.py:136
Function
load_state_dict
copy parameter from src_state_dict to model Arguments: model: A torch.nn.Module object src_state_dict: a dict containing para
baseline/utils/utils.py:216
Function
make_dir
(path)
script/dataset/transform_rap.py:10
Function
make_dir
(path)
script/dataset/transform_pa100k.py:10
Function
make_dir
(path)
script/dataset/transform_rap2.py:10
Function
make_dir
(path)
script/dataset/transform_peta.py:10
Method
reset
(self)
baseline/utils/utils.py:52
Method
reset
(self)
baseline/utils/utils.py:72
Method
reset
(self)
baseline/utils/utils.py:92
Function
resnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
baseline/model/resnet.py:193
Function
resnet152
Constructs a ResNet-152 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
baseline/model/resnet.py:205
Function
resnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
baseline/model/resnet.py:157
Function
resnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
baseline/model/resnet.py:169
Function
str2bool
(v)
baseline/utils/utils.py:16
Function
transfer_optims
(optims, device_id=-1)
baseline/utils/utils.py:193
Method
update
(self, val)
baseline/utils/utils.py:76
Method
update
(self, value)
baseline/utils/utils.py:96
Method
write
(self, msg)
baseline/utils/utils.py:143