MCPcopy Index your code

hub / github.com/ailia-ai/ailia-models / functions

Functions6,356 in github.com/ailia-ai/ailia-models

↓ 945 callersMethodreshape
Reshape inputs to expected shape.
diffusion/sd-turbo/df/pipelines/stable_diffusion_img2img.py:209
↓ 699 callersFunctioncheck_and_download_models
Check if the onnx file and prototxt file exists, and if necessary, download the files to the given path. Parameters ---------- w
util/model_utils.py:57
↓ 520 callersMethodresize
(self, image)
vision_language_model/mobilevlm/preprocessing_numpy.py:44
↓ 422 callersMethodcopy
Copies this config into a new config object, making the changes given by new_config_dict.
object_detection/yolact/yolact_util.py:85
↓ 398 callersFunctionget_base_parser
Get ailia default argument parser Parameters ---------- description : str default_input : str default input data (image
util/arg_utils.py:40
↓ 398 callersFunctionupdate_parser
Default check or update configurations should be placed here Parameters ---------- parser : ArgumentParser() Returns -------
util/arg_utils.py:124
↓ 367 callersMethodrun
(self, sde=None, logger=None, save_states=False)
diffusion/daclip-sde/util.py:172
↓ 357 callersMethodpredict
Propagate the state distribution to the current time step using a Kalman filter prediction step. Parameters ----------
object_tracking/deepsort/sort/track.py:109
↓ 343 callersMethodread
(self)
util/webcamera_utils.py:197
↓ 312 callersMethodclip
(self, width, height)
face_detection/dbface/dbface_utils.py:79
↓ 274 callersFunctionget_savepath
Get savepath NOTE: we may have better option... TODO: args.save_dir & args.save_path ? Parameters ---------- arg_path : str
util/arg_utils.py:219
↓ 179 callersMethodargmax
Get label with longest duration Parameters ---------- support : Segment or Timeline, optional Find label with lon
audio_processing/pyannote-audio/pyannote_audio_utils/core/annotation.py:1141
↓ 177 callersFunctionload_image
(image_path)
util/detector_utils.py:26
↓ 163 callersMethodreplace
Copies new_config_dict into this config object. Note: new_config_dict can also be a config object.
object_detection/yolact/yolact_util.py:98
↓ 139 callersFunctionimread
(filename, flags=cv2.IMREAD_COLOR)
util/image_utils.py:11
↓ 115 callersMethodpad
(self, *inputs)
optical_flow_estimation/raft/raft.py:127
↓ 108 callersMethodpredict
( self, features, orig_hw, point_coords: Optional[np.ndarray] = None,
image_segmentation/segment-anything-2/sam2_image_predictor.py:66
↓ 93 callersMethodpredict
Estimates the head pose given an image. Parameters ---------- img: NumPy array The image in BGR channels
face_recognition/hopenet/hopenet.py:92
↓ 92 callersFunctionplot_results
:param detector: ailia.Detector, or list of ailia.DetectorObject :param img: ndarray data of image :param category: list of category_name
util/detector_utils.py:100
↓ 88 callersMethodkeys
list of all keys
pose_estimation_3d/gast/libs/filterpy/common/helpers.py:161
↓ 85 callersMethodrun
(self, images: np.ndarray)
image_manipulation/docshadow/docshadow.py:62
↓ 82 callersFunctionget_capture
Get cv2.VideoCapture * TODO: maybe get capture & writer at the same time? * then, you can use capture frame size directory Pa
util/webcamera_utils.py:217
↓ 80 callersMethodflatten
Flattens any np.array of column vectors into 1D arrays. Basically, this makes data readable for humans if you are just inspecting via
pose_estimation_3d/gast/libs/filterpy/common/helpers.py:187
↓ 79 callersMethodfrom_pretrained
Instantiate a PreTrainedBertModel from a pre-trained model file. Download and cache the pre-trained model file if needed.
natural_language_processing/presumm/tokenizer.py:107
↓ 77 callersFunctionnormalize_image
Normalize image Parameters ---------- image: numpy array The image you want to normalize normalize_type: string
util/image_utils.py:20
↓ 73 callersMethodrun
(self,img,calib_file)
object_detection_3d/did_m3d/did_m3d_util.py:279
↓ 65 callersFunctionget_writer
get cv2.VideoWriter Parameters ---------- save_path : str height : int width : int fps : int rgb : bool, default is True
util/webcamera_utils.py:129
↓ 65 callersMethodresize
(images: np.ndarray, height, width)
diffusion/sd-turbo/df/pipelines/stable_diffusion_img2img.py:202
↓ 59 callersMethodfrom_pretrained
( cls, model_id: Union[str, Path], file_name: Optional[str], onnx: bool = Fals
diffusion/anything_v3/df/onnx_utils.py:154
↓ 56 callersMethodpop
(self, frame_no)
face_identification/arcface/arcface.py:176
↓ 54 callersMethodadd
Add a new hypothesis to the list.
vision_language_model/florence2/beam_search.py:252
↓ 53 callersMethodrun
(self, images, bboxes)
face_recognition/gazelle/gazelle_utils/model.py:49
↓ 50 callersFunctionload_image
Loads the image of the given path, performs the necessary preprocessing, and returns it. Parameters ---------- image_path: strin
util/image_utils.py:57
↓ 48 callersFunctioncheck_and_download_file
Check if the file exists, and if necessary, download the files to the given path. Parameters ---------- file_path: string
util/model_utils.py:92
↓ 48 callersFunctionwrite_predictions
(file_name, detector, img=None, category=None, file_type='txt')
util/detector_utils.py:211
↓ 46 callersFunctionsoftmax
(x, axis=None)
util/math_utils.py:6
↓ 44 callersMethodpredict
(self, word, encoder, decoder)
natural_language_processing/g2p_en/g2p_en.py:113
↓ 44 callersMethodsave
save the current state of the Kalman filter
pose_estimation_3d/gast/libs/filterpy/common/helpers.py:120
↓ 42 callersMethoddecode
Given a matrix of character probabilities, returns the decoder's best guess of the transcription Arguments: prob
audio_processing/deepspeech2/decoder.py:73
↓ 39 callersMethodpredict
Advances the state vector and returns the predicted bounding box estimate.
pose_estimation_3d/gast/gast_utils.py:165
↓ 35 callersFunctionline
(input_img, landmarks, flags, point1, point2)
pose_estimation_3d/blazepose-fullbody/blazepose-fullbody.py:143
↓ 35 callersMethodremove
Remove a segment (in place) Parameters ---------- segment : Segment Segment that is being removed Return
audio_processing/pyannote-audio/pyannote_audio_utils/core/timeline.py:279
↓ 34 callersMethodpredict
(self, input)
image_manipulation/invertible_denoising_network/invertible_denoising_network.py:52
↓ 33 callersMethodplot
(self, annotations, output_path, bboxes=None, points=None,
image_segmentation/fast_sam/prompt.py:193
↓ 32 callersMethodindex
Get index of (existing) segment Parameters ---------- segment : Segment Segment that is being looked for.
audio_processing/pyannote-audio/pyannote_audio_utils/core/timeline.py:224
↓ 31 callersMethodempty
(self)
object_tracking/qd-3dt/motion_tracker.py:76
↓ 30 callersMethodsearch
Search new tokens for running hypotheses and encoded speech x. Args: running_hyps (List[Hypothesis]): Running hypotheses on beam
audio_processing/reazon_speech/beam_search.py:164
↓ 29 callersFunctiondraw_connect
(from_id, to_id, color, X, Y, Z, IS_3D)
pose_estimation_3d/3d-pose-baseline/3d-pose-baseline.py:153
↓ 27 callersMethodcrop
Fast version of self(file).crop(segment, **kwargs) Parameters ---------- file : AudioFile Audio file. seg
audio_processing/pyannote-audio/pyannote_audio_utils/audio/core/io.py:248
↓ 26 callersFunctionpretty_str
Generates a pretty printed NumPy array with an assignment. Optionally transposes column vectors so they are drawn on one line. Strictly speak
pose_estimation_3d/gast/libs/filterpy/common/helpers.py:239
↓ 25 callersMethodlabels
Get sorted list of labels Returns ------- labels : list Sorted list of labels
audio_processing/pyannote-audio/pyannote_audio_utils/core/annotation.py:904
↓ 25 callersFunctionline
(input_img, landmarks, flags, point1, point2)
pose_estimation/blazepose/blazepose.py:68
↓ 24 callersFunctionprint_results
(classifier, labels, top_k=MAX_CLASS_COUNT)
util/classifier_utils.py:31
↓ 23 callersMethodcopy
Get a copy of the segment Returns ------- copy : Segment Copy of the segment.
audio_processing/pyannote-audio/pyannote_audio_utils/core/segment.py:195
↓ 23 callersMethodpredict
(self, image: Image)
style_transfer/psgan/psgan/preprocess.py:86
↓ 23 callersFunctionsigmoid
(x)
util/math_utils.py:14
↓ 22 callersFunctionplot_results
(input_image, classifier, labels, top_k=MAX_CLASS_COUNT, logging=True)
util/classifier_utils.py:50
↓ 21 callersMethodfrom_pretrained
Load pretrained pipeline Parameters ---------- checkpoint_path : Path or str Path to pipeline checkpoint, or a re
audio_processing/pyannote-audio/pyannote_audio_utils/audio/core/pipeline.py:46
↓ 21 callersFunctionresize
(img, size=None, out_shape=None, method='bilinear')
background_removal/cascade_psp/cascade_psp.py:62
↓ 21 callersMethodupdate
Perform Kalman filter measurement update step and update the feature cache. Parameters ---------- kf : kalman_filter.
object_tracking/deepsort/sort/track.py:123
↓ 20 callersMethodencode
(self, text, **kwargs)
audio_processing/whisper/tokenizer.py:42
↓ 20 callersFunctionline
(input_img, hand_keypoint, point1, point2, offset, scale)
commercial_model/acculus-pose/acculus-hand.py:58
↓ 18 callersFunctionline
(input_img, person, point1, point2)
action_recognition/ax_action_recognition/ax_action_recognition.py:168
↓ 18 callersFunctionline
(input_img, person, point1, point2)
pose_estimation_3d/3d-pose-baseline/3d-pose-baseline.py:283
↓ 18 callersFunctionline
(input_img, person, point1, point2)
commercial_model/acculus-pose/acculus-pose.py:62
↓ 18 callersFunctionline
(input_img, person, point1, point2)
pose_estimation/pose_resnet/pose_resnet.py:75
↓ 18 callersFunctionline
(input_img, person, point1, point2)
pose_estimation/openpose/openpose.py:77
↓ 18 callersFunctionline
(input_img, person, point1, point2)
pose_estimation/lightweight-human-pose-estimation/lightweight-human-pose-estimation.py:79
↓ 18 callersFunctionreverse_letterbox
(detections, img, det_shape)
util/detector_utils.py:74
↓ 18 callersMethodstft
(self, y)
audio_processing/voicefilter/audio_utils.py:80
↓ 17 callersMethodempty
Return an empty copy Returns ------- empty : Timeline Empty timeline using the same 'uri' attribute.
audio_processing/pyannote-audio/pyannote_audio_utils/core/timeline.py:695
↓ 16 callersMethodget_timeline
Get timeline made of all annotated segments Parameters ---------- copy : bool, optional Defaults (True) to return
audio_processing/pyannote-audio/pyannote_audio_utils/core/annotation.py:297
↓ 16 callersMethodpop
(self, *args, **kwargs)
diffusion/anything_v3/df/configuration_utils.py:48
↓ 16 callersFunctiontime_execution
(msg)
face_recognition/ax_gaze_estimation/ax_gaze_estimation.py:130
↓ 15 callersFunctionC_SUB
(res, a, b)
audio_processing/rnnoise/kiss_fft.py:32
↓ 15 callersFunctionget_image_crop_size
()
anomaly_detection/padim/padim_gui.py:197
↓ 15 callersFunctionline
(input_img, coordinates, point1, point2)
pose_estimation/efficientpose/efficientpose.py:84
↓ 15 callersMethodplot
(self, img, vertices, edges)
pose_estimation_3d/lightweight-human-pose-estimation-3d/modules/draw.py:36
↓ 15 callersMethodtransform
(self, data)
time_series_forecasting/informer2020/informer2020_utils.py:18
↓ 14 callersFunctionadd
(name, *args)
natural_language_processing/g2p_en/averaged_perceptron.py:97
↓ 14 callersMethodfit
Estimates a box 9-dof parameters from the given vertices. Directly computes the scale of the box, then solves for orientation and tra
object_detection_3d/mediapipe_objectron/objectron/dataset/box.py:123
↓ 14 callersMethodforward
(self, x)
audio_processing/vall-e-x/modules/embedding.py:25
↓ 14 callersMethoditertracks
Iterate over tracks (in chronological order) Parameters ---------- yield_label : bool, optional When True, yield
audio_processing/pyannote-audio/pyannote_audio_utils/core/annotation.py:262
↓ 14 callersFunctionline
(input_img, person, point1, point2)
commercial_model/acculus-pose/acculus-up-pose.py:71
↓ 14 callersFunctionnms_boxes
(boxes, scores, iou_thres)
util/nms_utils.py:58
↓ 14 callersMethodpop
(self, *args, **kwargs)
diffusion/anything_v3/df/utils/outputs.py:64
↓ 13 callersMethoddraw
Draws the head pose(s) on the image. (Person POV) The axes correspond to x (blue) oriented positively to the left, y (green) oriented
face_recognition/hopenet/hopenet.py:159
↓ 13 callersFunctionnormalize
(x)
neural_rendering/nerf/load_llff.py:16
↓ 13 callersMethodnormalize
(self, S)
audio_processing/voicefilter/audio_utils.py:99
↓ 13 callersFunctionout_fmt
(input, cur_id_b, cur_id_self, cur_id_a, ply_flag)
image_inpainting/3d-photo-inpainting/trid_photo_inpainting_utils.py:2312
↓ 13 callersMethodpredict
(self, track1, track2)
object_tracking/strong_sort/AppFreeLink.py:78
↓ 13 callersMethodupdate
(self, tokens, logits, sum_logprobs, rearrange_kv_cache)
audio_processing/whisper/decode_utils.py:54
↓ 12 callersMethodadd
Add a segment (in place) Parameters ---------- segment : Segment Segment that is being added Returns
audio_processing/pyannote-audio/pyannote_audio_utils/core/timeline.py:243
↓ 12 callersFunctionbatched_nms
(boxes, scores, labels, iou_thres)
util/nms_utils.py:81
↓ 12 callersFunctionline
(img,p1, p2)
object_detection_3d/did_m3d/did_m3d.py:52
↓ 12 callersFunctionload_detail
(image_path, is_train)
anomaly_detection/padim/padim_gui.py:173
↓ 11 callersMethodconvert_ids_to_tokens
Converts a sequence of ids in wordpiece tokens using the vocab.
natural_language_processing/presumm/tokenizer.py:99
↓ 11 callersMethodload_onnx
(self, net)
audio_processing/vall-e-x/modules/embedding.py:32
↓ 11 callersMethodpredict
Predict the gaze given an image. Parameters ---------- img : NumPy array The image in BGR channels. gazes
face_recognition/ax_gaze_estimation/ax_gaze_estimation.py:175
↓ 11 callersMethodpredict
(self)
face_recognition/mivolo/tracker/byte_tracker.py:21
next →1–100 of 6,356, ranked by callers