File size: 929 Bytes
92e0882 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
"""A naive baseline method: just pass the full expression to CLIP."""
from overrides import overrides
from typing import Dict, Any
import random
from argparse import Namespace
import numpy as np
from .ref_method import RefMethod
class Random(RefMethod):
"""CLIP-only baseline where each box is evaluated with the full expression."""
def __init__(self, args: Namespace):
self.box_area_threshold = args.box_area_threshold
@overrides
def execute(self, caption: str, env: "Environment") -> Dict[str, Any]:
probs = env.filter_area(self.box_area_threshold)*env.uniform()
random_ordering = list(range(len(env.boxes)))
random.shuffle(random_ordering)
random_ordering = np.array(random_ordering)
pred = np.argmax(probs*random_ordering)
return {
"probs": probs.tolist(),
"pred": int(pred),
"text": caption.lower()
}
|