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:
|