search

Event-AHU / OpenPAR

[OpenPAR] An open-source framework for Pedestrian Attribute Recognition, based on PyTorch
MIT License
63 stars 6 forks source link

How to get Prediction on a single image. using Promptpar #27

Closed asad1416 closed 1 week ago

asad1416 commented 2 weeks ago

I am trying to get prediction on a single image, test_example.py is running but I could not find model.predict funciton. Can you please guide me how I can get prediction in the form of attributes or token and how we can get a sentence like output.

1125178969 commented 2 weeks ago

You can perform a forward pass and then sigmoid the model's predictions, and then output the index of the corresponding attribute list through threshold judgment