Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dddzg/up-detr
/ functions
Functions
193 in github.com/dddzg/up-detr
⨍
Functions
193
◇
Types & classes
43
↓ 29 callers
Function
print
(*args, **kwargs)
util/misc.py:381
↓ 16 callers
Method
to
(self, device)
util/misc.py:300
↓ 9 callers
Method
max
(self)
util/misc.py:78
↓ 9 callers
Method
update
(self, **kwargs)
util/misc.py:169
↓ 6 callers
Function
nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:318
↓ 6 callers
Method
with_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:216
↓ 4 callers
Method
__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
models/transformer.py:26
↓ 4 callers
Method
__init__
(self, detr, freeze_detr=False)
models/segmentation.py:31
↓ 4 callers
Function
_expand
(tensor, length: int)
models/segmentation.py:72
↓ 4 callers
Method
_get_src_permutation_idx
(self, indices)
models/detr.py:220
↓ 4 callers
Method
decompose
(self)
util/misc.py:311
↓ 4 callers
Function
is_dist_avail_and_initialized
()
util/misc.py:389
↓ 3 callers
Method
__init__
(self, backbone, position_embedding)
models/backbone.py:114
↓ 3 callers
Function
_run
(command)
util/misc.py:257
↓ 3 callers
Method
add_meter
(self, name, meter)
util/misc.py:196
↓ 3 callers
Function
build_dataset
(image_set, args)
datasets/__init__.py:25
↓ 3 callers
Function
crop
(image, target, region)
datasets/transforms.py:22
↓ 3 callers
Function
get_world_size
()
util/misc.py:397
↓ 3 callers
Function
interpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and t
util/misc.py:462
↓ 2 callers
Method
__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/detr.py:321
↓ 2 callers
Function
_get_activation_fn
Return an activation function given a string
models/transformer.py:296
↓ 2 callers
Function
_get_clones
(module, N)
models/transformer.py:279
↓ 2 callers
Method
_set_aux_loss
(self, outputs_class, outputs_coord, outputs_feature, backbone_output)
models/updetr.py:131
↓ 2 callers
Method
_set_aux_loss
(self, outputs_class, outputs_coord)
models/detr.py:81
↓ 2 callers
Function
all_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
util/misc.py:94
↓ 2 callers
Function
box_cxcywh_to_xyxy
(x)
util/box_ops.py:15
↓ 2 callers
Function
build_model
(args)
models/__init__.py:18
↓ 2 callers
Function
convert_to_xywh
(boxes)
datasets/coco_eval.py:171
↓ 2 callers
Function
evaluate
(model, criterion, postprocessors, data_loader, base_ds, device, output_dir)
engine.py:75
↓ 2 callers
Function
get_coco_api_from_dataset
(dataset)
datasets/__init__.py:15
↓ 2 callers
Method
get_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
models/detr.py:232
↓ 2 callers
Method
log_every
(self, iterable, print_freq, header=None)
util/misc.py:199
↓ 2 callers
Function
make_coco_transforms
(image_set)
datasets/coco.py:121
↓ 2 callers
Method
prepare
(self, predictions, iou_type)
datasets/coco_eval.py:75
↓ 2 callers
Method
synchronize_between_processes
(self)
util/misc.py:192
↓ 2 callers
Function
train_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Iterable, optimizer: tor
engine.py:23
↓ 2 callers
Method
update
(self, value, n=1)
util/misc.py:45
↓ 2 callers
Method
with_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:153
↓ 1 callers
Method
__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
models/position_encoding.py:23
↓ 1 callers
Method
_get_tgt_permutation_idx
(self, indices)
models/detr.py:226
↓ 1 callers
Function
_max_by_axis
(the_list)
util/misc.py:286
↓ 1 callers
Function
_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:346
↓ 1 callers
Method
_reset_parameters
(self)
models/transformer.py:48
↓ 1 callers
Method
accumulate
(self)
datasets/coco_eval.py:66
↓ 1 callers
Function
accuracy
Computes the precision@k for the specified values of k
util/misc.py:444
↓ 1 callers
Function
box_iou
(boxes1, boxes2)
util/box_ops.py:30
↓ 1 callers
Function
box_xyxy_to_cxcywh
(x)
util/box_ops.py:22
↓ 1 callers
Function
build_backbone
(args)
models/backbone.py:134
↓ 1 callers
Function
build_matcher
(args)
models/matcher.py:91
↓ 1 callers
Function
build_position_encoding
(args)
models/position_encoding.py:85
↓ 1 callers
Function
build_selfdet
(image_set, args)
datasets/selfdet.py:148
↓ 1 callers
Function
build_transformer
(args)
models/transformer.py:283
↓ 1 callers
Function
convert_coco_poly_to_mask
(segmentations, height, width)
datasets/coco.py:39
↓ 1 callers
Function
create_common_coco_eval
(coco_eval, img_ids, eval_imgs)
datasets/coco_eval.py:198
↓ 1 callers
Function
dice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
models/segmentation.py:180
↓ 1 callers
Function
evaluate
Run per image evaluation on given images and store results (a list of dict) in self.evalImgs :return: None
datasets/coco_eval.py:214
↓ 1 callers
Method
forward_post
(self, src, src_mask: Optional[Tensor] = None,
models/transformer.py:156
↓ 1 callers
Method
forward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
models/transformer.py:219
↓ 1 callers
Method
forward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
models/transformer.py:171
↓ 1 callers
Method
forward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
models/transformer.py:242
↓ 1 callers
Function
generalized_box_iou
Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where
util/box_ops.py:46
↓ 1 callers
Function
get_args_parser
()
detr_main.py:26
↓ 1 callers
Function
get_args_parser
()
main.py:25
↓ 1 callers
Function
get_query_transforms
(image_set)
datasets/selfdet.py:122
↓ 1 callers
Function
get_random_patch_from_img
:param img: original image :param min_pixel: min pixels of the query patch :return: query_patch,x,y,w,h
datasets/selfdet.py:18
↓ 1 callers
Function
get_rank
()
util/misc.py:403
↓ 1 callers
Function
get_size
(image_size, size, max_size=None)
datasets/transforms.py:105
↓ 1 callers
Function
get_size_with_aspect_ratio
(image_size, size, max_size=None)
datasets/transforms.py:85
↓ 1 callers
Function
hflip
(image, target)
datasets/transforms.py:65
↓ 1 callers
Function
is_main_process
()
util/misc.py:409
↓ 1 callers
Function
main
(args)
detr_main.py:112
↓ 1 callers
Function
main
(args)
main.py:104
↓ 1 callers
Function
make_self_det_transforms
(image_set)
datasets/selfdet.py:83
↓ 1 callers
Function
masks_to_boxes
Compute the bounding boxes around the provided masks The masks should be in format [N, H, W] where N is the number of masks, (H, W) are the spati
util/box_ops.py:70
↓ 1 callers
Function
merge
(img_ids, eval_imgs)
datasets/coco_eval.py:176
↓ 1 callers
Function
pad
(image, target, padding)
datasets/transforms.py:141
↓ 1 callers
Method
prepare_for_coco_detection
(self, predictions)
datasets/coco_eval.py:85
↓ 1 callers
Method
prepare_for_coco_keypoint
(self, predictions)
datasets/coco_eval.py:144
↓ 1 callers
Method
prepare_for_coco_segmentation
(self, predictions)
datasets/coco_eval.py:109
↓ 1 callers
Method
reset_parameters
(self)
models/position_encoding.py:67
↓ 1 callers
Function
resize
(image, target, size, max_size=None)
datasets/transforms.py:82
↓ 1 callers
Function
setup_for_distributed
This function disables printing when not in master process
util/misc.py:374
↓ 1 callers
Function
sigmoid_focal_loss
Loss used in RetinaNet for dense detection: https://arxiv.org/abs/1708.02002. Args: inputs: A float tensor of arbitrary shape.
models/segmentation.py:198
↓ 1 callers
Method
summarize
(self)
datasets/coco_eval.py:70
↓ 1 callers
Method
summarize
(self)
datasets/panoptic_eval.py:43
↓ 1 callers
Method
synchronize_between_processes
(self)
datasets/coco_eval.py:61
↓ 1 callers
Method
synchronize_between_processes
(self)
datasets/panoptic_eval.py:36
↓ 1 callers
Method
update
(self, predictions)
datasets/coco_eval.py:42
↓ 1 callers
Method
update
(self, predictions)
datasets/panoptic_eval.py:29
Method
__call__
(self, img, target)
datasets/transforms.py:158
Method
__call__
(self, img: PIL.Image.Image, target: dict)
datasets/transforms.py:168
Method
__call__
(self, img, target)
datasets/transforms.py:179
Method
__call__
(self, img, target)
datasets/transforms.py:191
Method
__call__
(self, img, target=None)
datasets/transforms.py:203
Method
__call__
(self, img, target)
datasets/transforms.py:212
Method
__call__
(self, img, target)
datasets/transforms.py:228
Method
__call__
(self, img, target)
datasets/transforms.py:235
Method
__call__
(self, img, target)
datasets/transforms.py:244
Method
__call__
(self, image, target=None)
datasets/transforms.py:253
Method
__call__
(self, image, target)
datasets/transforms.py:271
next →
1–100 of 193, ranked by callers