project-sulsul / sulsul-ai Goto Github PK

Features

• Multi-Label Classification

  

• Image Padding

• Quantization Aware Training

• training

  python3 train.py --data_path 'your/dataset/directory' --name 'prj' --model 'resnet18' --pretrained --num_classes 39 --batch_size 32 --lr_scheduling --check_point
  

• test

  python3 evaluate.py --data_path 'your/dataset/directory' --quantization 'qat' --model_name 'resnet18' --weight 'your/best/weight/directory' --num_classes 39
  

• inference

  python3 inference.py --img_url 'image/url' --model_name 'resnet18' --quantization 'qat' --weight 'your/best/weight/directory' --num_classes 39 --threshold 0.5