brails.processors.vlm_image_classifier.clip.utils module

brails.processors.vlm_image_classifier.clip.utils.aggregate_predictions(similarity, agg_method, gap)

brails.processors.vlm_image_classifier.clip.utils.compute_similarity(image_features, text_features)

brails.processors.vlm_image_classifier.clip.utils.pred_idx_to_labels(gt_labels, prediction_df)

brails.processors.vlm_image_classifier.clip.utils.predict_wrapper(model, text_input, img_paths, preprocess, device, batch_size=200, agg='max', num_classes=3)

brails.processors.vlm_image_classifier.clip.utils.preprocess_batch_img(batch_idx, batch_size, img_paths, aug, device)

brails.processors.vlm_image_classifier.clip.utils.round(similarity, decimals)