Unverified Commit c9cc3178 authored by echo840's avatar echo840 Committed by GitHub
Browse files

Update readme.md

parent a5c9ff66
...@@ -15,7 +15,7 @@ Generation Steps: ...@@ -15,7 +15,7 @@ Generation Steps:
```python grit_generate.py``` ```python grit_generate.py```
3. Generate segmentation maps for each image using the SAM model, and save the segmentation maps in the "./masks" directory. 3. Generate segmentation maps for each image using the SAM model, and save the segmentation maps in the "./masks" directory.
```python amg.py --checkpoint ./model_weight/<pth name> --model-type <model_type> --input ./images --output ./data_gen/masks --convert-to-rle``` ```python amg.py --checkpoint ./model_weight/<pth name> --model-type <model_type> --input ./images --output ./masks --convert-to-rle```
4. Generate corresponding descriptions for the segmentation maps. 4. Generate corresponding descriptions for the segmentation maps.
```python sam_blip.py``` ```python sam_blip.py```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment