search

yuhuan-wu / JCS

[IEEE TIP 2021] COVID-CS Dataset and Code of JCS: An explainable COVID-19 diagnosis system by joint classification and segmentation
48 stars 13 forks source link

Training process #1

Closed huanghjA closed 3 years ago

huanghjA commented 3 years ago

Hi, thanks for your share of JCS model. How can I access the training code?

yuhuan-wu commented 3 years ago

Thanks for your attention. The training code is still under preparation. If available, in this week, I will release the training code.

tifat58 commented 3 years ago

Hi, Any update on training code? is it possible to release it soon. Thanks

yuhuan-wu commented 3 years ago

Hi, I have uploaded the training examples to this repository. Plz run git pull and start to train our model with training examples:

CUDA_VISIBLE_DEVICES=0 ./tools/train.sh

tifat58 commented 3 years ago

Thanks a lot :)

yuhuan-wu commented 3 years ago

It should not have training issues now. I decide to close this issue.

If you have any other questions, please discuss via creating a new issue.

Thanks :)