Update README.md

README.md

@@ -16,7 +16,7 @@ TorchVision 0.14.1
 
 ### Datasets
 We follow the data set structure of HQSeg-44K as follows：
-'''
+```
 data
 |____DIS5K
 |____cascade_psp
@@ -28,18 +28,18 @@ data
 |____thin_object_detection
 | |____COIFT
 | |____ThinObject5K
-'''
+```
 You can get the datasets from [here](https://drive.google.com/drive/folders/1j1yFEejTAdAQzSbCrdBWoHE4VjaAf25L?usp=drive_link)
 ### Train
-'''
+```
 python -m torch.distributed.launch --nproc_per_node=<num_gpus> train.py --checkpoint <your checkpoint path> --model-type <model_type> --output <your output path>
-'''
+```
 
 EG-SAM is an improvement on the basis of HQ-SAM, you can follow the environment Settings [here](https://github.com/SysCV/SAM-HQ?tab=readme-ov-file)
 ### Evaluation
-'''
+'```
 python -m torch.distributed.launch --nproc_per_node=<num_gpus> train.py --checkpoint <your checkpoint path> --model-type <model_type> --output <your output path> --eval --restore-model <your training_checkpoint path>
-'''
+```
 You can get the weight file [here](https://drive.google.com/file/d/1B9-bTQ4c_fG8s--837HpMhaUT5gWGdv9/view?usp=drive_link)
 
 ### Visualization