RMB value classification & currency code detection and recognization
训练数据、测试数据、csv标签、模型分别放入以下路径:
train_data_path = '../../dataset_warm_up/train_data/'
test_data_path = '../../dataset_warm_up/public_test_data/'
csv_file = '../../dataset_warm_up/train_face_value_label.csv'
model_saved_path = '../../dataset_warm_up/saved_model/'
即可训练。
-
pytorch 0.4
-
matplotlib 1.5.3
-
tqdm 4.11.2
-
sklearn 0.19.1
-
pandas 0.20.3
-
PIL 4.2.1
-
torchvision 0.2.1
-
numpy 1.16.0