Tongue Size and Shape Classification Fusing Segmentation Features for Traditional Chinese Medicine Diagnosis
by Yating Huang+, Xuechen Li+, Siting Zheng, Zhonglian Li, Sihan Li, Linlin Shen, Changen Zhou, Zhihui Lai.(+ indicates equal contribution)
This repository is for our paper"Tongue Size and Shape Classification Fusing Segmentation Features for Traditional Chinese Medicine Diagnosis"
Ubuntu 16.04+pycharm+python3.6+pytorch1.7.1
Clone the repository:
git clone https://github.com/Yating-Huang/TSC-WNet.git
cd TSC-WNet
The only thing you should do is enter the data/image.txt and correct the path of the datasets. then run ~ example:
python kfold_class.py
After training, the saved model is in this folder.
the public Tongue dataset link:https://github.com/BioHit/TongeImageDataset
- The repository is being updated
- Contact: Yating Huang ([email protected])