"OneClassification" is one label image classification.
- configs: for dataset choice configuration
- work_dirs: for the save directory of training result
-
set your data set name, root, annotations, image directory
data_name = 'your_data_set_name'
data_root = 'your_data_set_root'
raw_train_path = 'your_data_annotations'
ann_file = 'your_data_annotations'
img_prefix = 'your_image_directory'
-
start training
python train.py
dataset name score
pneumonia 78.0543
dofu_tomato 96.9436
This project is licensed under the MIT license.