File size: 1,117 Bytes
0522016 | 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 | import json, os, sys, time
HERE = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, os.path.join(HERE, "..", "rk3588"))
EVAL = os.environ.get("EVAL_DIR", os.path.join(HERE, "eval"))
from PIL import Image
import locate3b as L
S = json.load(open(EVAL + "/eval_set.json"))
loc = L.Locator()
out = {}
t0 = time.time()
for k, s in enumerate(S):
img = Image.open(EVAL + "/" + s["file"])
q = "Locate all the instances that matches the following description: %s." % s["cat"]
t = time.time()
try:
text, st = loc.locate(img, q, max_new=400, rep_penalty=1.1, max_boxes=100, dedupe_iou=0.85, temperature=0.7, top_p=0.9, seed=0)
except Exception as e:
text, st = "ERROR %r" % e, {}
out[str(s["id"])] = {"text": text, "sec": round(time.time() - t, 2), "stats": st}
if k % 10 == 0:
print(k, "%.0fs" % (time.time() - t0), flush=True)
json.dump(out, open(EVAL + "/preds_board_sampled.json", "w"), ensure_ascii=False)
json.dump(out, open(EVAL + "/preds_board_sampled.json", "w"), ensure_ascii=False)
loc.close(); print("BOARD DONE", "%.0fs" % (time.time() - t0))
|