Skip to content
Snippets Groups Projects

Add start_token parameter to prediction function

Merged Mélodie Boillet requested to merge start-token-predict into main
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
+ 1
1
@@ -6,9 +6,9 @@ import re
import cv2
import numpy as np
import torch
import yaml
from dan import logger
from dan.datasets.extract.utils import save_json
from dan.decoder import GlobalHTADecoder
Loading