🏆 The third prize of the 14th National College Student Software Innovation Competition.
- Put the the annotated dataset in
data
directory. - run
python -m src.preprocess.peanut
We have annotated part of the Chinese handwriting character dataset, please refer to ccsa for more details.
Run base network
python -m src.preprocess.run 'train' 'base' --batch-size=16 --num-workers=0 --epoch=50 --step-size=5 --lr=3e-4 --use-gpu --gpu=0 --apex
Run refinement network
python -m src.preprocess.run 'train' 'refinement' --batch-size=16 --num-workers=0 --epoch=10 --step-size=5 --lr=5e-3 --use-gpu --gpu=0 --apex
Run classification network to get a pretrained model
python -m src.preprocess.run 'train' 'cls' --batch-size=16 --num-workers=0 --epoch=10 --step-size=5 --lr=5e-3 --use-gpu --gpu=0 --apex
Run inference model, using --snapshot
to specify model weights
python -m src.preprocess.run 'test' 'openpose' --batch-size=32 --num-workers=0 --use-gpu --gpu=2 --apex --reinforcement --snapshot='./checkpoints/openpose_model_apex_ep2.pt'
python -m src.preprocess.run 'view' 'base'
Heatmap Results
Character Structure Detection