Open ohjho opened 3 months ago
added a new method inference() to unisal.train.Trainer and a new module unisal.demo so that inference on an single image using the trained model can be as simple as:
inference()
import unisal saliency_map = unisal.demo.predict_image(img_rgb=np.array(pil_im))
for a more detailed usage, see the function predict_image() in run.py
predict_image()
support for video inference will come in a future PR
added a new method
inference()
to unisal.train.Trainer and a new module unisal.demo so that inference on an single image using the trained model can be as simple as:for a more detailed usage, see the function
predict_image()
in run.pysupport for video inference will come in a future PR