Example 12: Train all the internal reider classifiers for GTA V dataset

• Description: Use the standalone trainReIDClassifier() to train and save a classifier .PKL files for GTA V dataset.

• Featuring:

  • trainReIDClassifier() | pyppbox.standalone.trainReIDClassifier()

ℹ️ Source code and input file(s) -> {pyppbox repo}/examples

#################################################################################
# Example of retraining all the internal reider classifiers for GTA V dataset
#################################################################################

import os
import pyppbox
from pyppbox.standalone import trainReIDClassifier


pyppbox_root = os.path.dirname(pyppbox.__file__)

gta5_osnet_ain_config = {
    'ri_name': 'Torchreid', 
    'classifier_pkl': f'{pyppbox_root}/data/modules/torchreid/classifier/gta5_osnet_ain_ms_d_c.pkl', 
    'train_data': f'{pyppbox_root}/data/datasets/GTA_V_DATASET/body_128x256', 
    'model_name': 'osnet_ain_x1_0', 
    'model_path': f'{pyppbox_root}/data/modules/torchreid/models/torchreid/osnet_ain_ms_d_c.pth.tar', 
    'min_confidence': 0.35, 
    'device': 'cuda'
}

gta5_osnet_config = {
    'ri_name': 'Torchreid', 
    'classifier_pkl': f'{pyppbox_root}/data/modules/torchreid/classifier/gta5_osnet_ms_d_c.pkl', 
    'train_data': f'{pyppbox_root}/data/datasets/GTA_V_DATASET/body_128x256', 
    'model_name': 'osnet_x1_0', 
    'model_path': f'{pyppbox_root}/data/modules/torchreid/models/torchreid/osnet_ms_d_c.pth.tar', 
    'min_confidence': 0.35, 
    'device': 'cuda'
}

gta5_facenet_config = {
    'ri_name': 'FaceNet', 
    'gpu_mem': 0.585, 
    'model_det': f'{pyppbox_root}/data/modules/facenet/models/det', 
    'model_file': f'{pyppbox_root}/data/modules/facenet/models/20180402-114759/20180402-114759.pb', 
    'classifier_pkl': f'{pyppbox_root}/data/modules/facenet/classifier/gta5.pkl', 
    'train_data': f'{pyppbox_root}/data/datasets/GTA_V_DATASET/face_182x182', 
    'batch_size': 1000, 
    'min_confidence': 0.75, 
    'yl_h_calibration': [-125, 75], 
    'yl_w_calibration': [-55, 55]
}

trainReIDClassifier(reider=gta5_osnet_ain_config)
trainReIDClassifier(reider=gta5_osnet_config)
trainReIDClassifier(reider=gta5_facenet_config)