You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Sorry for the terribly late response. If I understand your intention correctly, you are trying to do inference using the model. Unfortunately, the code has no proper support for inference mode yet. It's actually not hard to make this work based on the existing codebase as we are using PyTorch.
Steps on how to achieve this:
Create a new function in main.py and named it infer. Make this function accept a new input parameter, data for passing in images from your dataset (0001.jpg, 0002.jpg, etc.)
Convert the pixel data from the image file into multi-dimensional array/vector.
Can you tell us how can we use your code to classify RGB images.
Our dataset is like that:
class1:
0001.jpg
0002.jpg
Class2:
001.jpg
002.jpg
The text was updated successfully, but these errors were encountered: