Update README.md
Browse files
README.md
CHANGED
|
@@ -25,6 +25,7 @@ cd mmsegmentation
|
|
| 25 |
pip install -v -e .
|
| 26 |
```
|
| 27 |
***Creating a Dataset***
|
|
|
|
| 28 |
|
| 29 |
1. Directory structure of the dataset:
|
| 30 |
<pre>
|
|
@@ -63,7 +64,7 @@ pip install -v -e .
|
|
| 63 |
***Dataset Configs***
|
| 64 |
|
| 65 |
```bash
|
| 66 |
-
cd mmseg/datasets
|
| 67 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/dataset_configs/CVRP.py
|
| 68 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/dataset_configs/__init__.py
|
| 69 |
cd ../../configs/_base_/datasets
|
|
@@ -109,17 +110,17 @@ You can generate model config files using *run_configs.py*
|
|
| 109 |
|
| 110 |
```bash
|
| 111 |
mkdir 'work_dirs' 'CVRP_configs' 'outputs'
|
| 112 |
-
python run/run_configs.py --model_name deeplabv3plus -m configs/deeplabv3plus/deeplabv3plus_r101-d8_4xb4-160k_ade20k-512x512.py -d configs/_base_/datasets/CVRP_pipeline.py -c 2 -w work_dirs -s CVRP_configs
|
| 113 |
|
| 114 |
-
python run/run_configs.py --model_name knet -m configs/knet/knet-s3_swin-l_upernet_8xb2-adamw-80k_ade20k-512x512.py -d configs/_base_/datasets/CVRP_pipeline.py -c 2 -w work_dirs -s CVRP_configs
|
| 115 |
|
| 116 |
-
python run/run_configs.py --model_name mask2former -m configs/mask2former/mask2former_swin-l-in22k-384x384-pre_8xb2-160k_ade20k-640x640.py -d configs/_base_/datasets/CVRP_pipeline.py -c 2 -w work_dirs -s CVRP_configs
|
| 117 |
|
| 118 |
-
python run/run_configs.py --model_name segformer -m configs/segformer/segformer_mit-b5_8xb2-160k_ade20k-512x512.py -d configs/_base_/datasets/CVRP_pipeline.py -c 2 -w work_dirs -s CVRP_configs
|
| 119 |
```
|
| 120 |
Also, you can download model config files [here](https://huggingface.co/CVRPDataset/Model/tree/main/model_configs).
|
| 121 |
-
```bash
|
| 122 |
-
cd CVRP_configs
|
| 123 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/model_configs/CVRP_mask2former.py
|
| 124 |
```
|
| 125 |
|
|
@@ -133,7 +134,7 @@ Also, you can download checkpoint [here](https://huggingface.co/CVRPDataset/Mode
|
|
| 133 |
***Test***
|
| 134 |
```bash
|
| 135 |
cd ..
|
| 136 |
-
python run/test.py -d CVRPDataset/val -m CVRP_configs/CVRP_mask2former.py -pth work_dirs/CVRP_mask2former/Mask2Former.pth -o outputs/CVRP_mask2former
|
| 137 |
```
|
| 138 |
### UI
|
| 139 |
##### We create a web user interface for annotation based on gradio:
|
|
|
|
| 25 |
pip install -v -e .
|
| 26 |
```
|
| 27 |
***Creating a Dataset***
|
| 28 |
+
Here is an example if you want to make own dataset.
|
| 29 |
|
| 30 |
1. Directory structure of the dataset:
|
| 31 |
<pre>
|
|
|
|
| 64 |
***Dataset Configs***
|
| 65 |
|
| 66 |
```bash
|
| 67 |
+
cd mmsegmentation/mmseg/datasets
|
| 68 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/dataset_configs/CVRP.py
|
| 69 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/dataset_configs/__init__.py
|
| 70 |
cd ../../configs/_base_/datasets
|
|
|
|
| 110 |
|
| 111 |
```bash
|
| 112 |
mkdir 'work_dirs' 'CVRP_configs' 'outputs'
|
| 113 |
+
python run/run_configs.py --model_name deeplabv3plus -m mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r101-d8_4xb4-160k_ade20k-512x512.py -d mmsegmentation/configs/_base_/datasets/CVRP_pipeline.py -c 2 -w mmsegmentation/work_dirs -s mmsegmentation/CVRP_configs
|
| 114 |
|
| 115 |
+
python run/run_configs.py --model_name knet -m mmsegmentation/configs/knet/knet-s3_swin-l_upernet_8xb2-adamw-80k_ade20k-512x512.py -d mmsegmentation/configs/_base_/datasets/CVRP_pipeline.py -c 2 -w mmsegmentation/work_dirs -s mmsegmentation/CVRP_configs
|
| 116 |
|
| 117 |
+
python run/run_configs.py --model_name mask2former -m mmsegmentation/configs/mask2former/mask2former_swin-l-in22k-384x384-pre_8xb2-160k_ade20k-640x640.py -d mmsegmentation/configs/_base_/datasets/CVRP_pipeline.py -c 2 -w mmsegmentation/work_dirs -s mmsegmentation/CVRP_configs
|
| 118 |
|
| 119 |
+
python run/run_configs.py --model_name segformer -m mmsegmentation/configs/segformer/segformer_mit-b5_8xb2-160k_ade20k-512x512.py -d mmsegmentation/configs/_base_/datasets/CVRP_pipeline.py -c 2 -w mmsegmentation/work_dirs -s mmsegmentation/CVRP_configs
|
| 120 |
```
|
| 121 |
Also, you can download model config files [here](https://huggingface.co/CVRPDataset/Model/tree/main/model_configs).
|
| 122 |
+
```bash
|
| 123 |
+
cd mmsegmentation/CVRP_configs
|
| 124 |
wget https://huggingface.co/CVRPDataset/Model/resolve/main/model_configs/CVRP_mask2former.py
|
| 125 |
```
|
| 126 |
|
|
|
|
| 134 |
***Test***
|
| 135 |
```bash
|
| 136 |
cd ..
|
| 137 |
+
python run/test.py -d mmsegmentation/CVRPDataset/val -m mmsegmentation/CVRP_configs/CVRP_mask2former.py -pth mmsegmentation/work_dirs/CVRP_mask2former/Mask2Former.pth -o mmsegmentation/outputs/CVRP_mask2former
|
| 138 |
```
|
| 139 |
### UI
|
| 140 |
##### We create a web user interface for annotation based on gradio:
|