Closed cuge1995 closed 3 years ago
The original class index is in dictionary order, while our index depends on the order of function os.listdir()
. If you wanted to load our pretrained model, the referred type_to_index_map
should be used. Otherwise, you could use any order of class index.
but in your code: