diff --git a/README.md b/README.md index b7899a3..446f058 100644 --- a/README.md +++ b/README.md @@ -39,6 +39,12 @@ Then you can modify your gpu lists in config.py for specified gpus and then run: ``` python train.py ``` +You can visualize your training steps via tensorboard: +``` +tensorboard --logdir='./checkpoint' +``` +![](imgs/loss.png) + Use ICDAR2019-LSVT, ICDAR2019-ArT, ICDAR2019-ReCTS for default training, you can change it with your own training data. ### Evaluation diff --git a/imgs/loss.png b/imgs/loss.png new file mode 100644 index 0000000..e9e423f Binary files /dev/null and b/imgs/loss.png differ