Attention
The model file is in the .pth format and the configuration file is in the .json format. The input data to the Image Classification Step should be consistent with the model training data.
Figure 1. The Image Classification Step in Mech-Vision¶