Skip to content

Commit 078a96f

Browse files
update README
1 parent de1ce80 commit 078a96f

File tree

2 files changed

+12
-16
lines changed

2 files changed

+12
-16
lines changed

README.md

+11-15
Original file line numberDiff line numberDiff line change
@@ -81,9 +81,7 @@ All your training image will be scaled into the same scale according to the conf
8181
Use the script here to generate the tensorflow records file
8282

8383
```
84-
python data_provider/lanenet_data_feed_pipline.py
85-
--dataset_dir ./data/training_data_example
86-
--tfrecords_dir ./data/training_data_example/tfrecords
84+
python tools/make_tusimple_tfrecords.py
8785
```
8886

8987
#### Train model
@@ -95,18 +93,7 @@ script to load your own pretrained parameters or you can implement your own base
9593
You may call the following script to train your own model
9694

9795
```
98-
python tools/train_lanenet.py
99-
--net vgg
100-
--dataset_dir ./data/training_data_example
101-
-m 0
102-
```
103-
You can also continue the training process from the snapshot by
104-
```
105-
python tools/train_lanenet.py
106-
--net vgg
107-
--dataset_dir data/training_data_example/
108-
--weights_path path/to/your/last/checkpoint
109-
-m 0
96+
python tools/train_lanenet_tusimple.py
11097
```
11198

11299
You may monitor the training process using tensorboard tools
@@ -171,6 +158,15 @@ script on your own.
171158

172159
New model weights can be found [here](https://www.dropbox.com/sh/tnsf0lw6psszvy4/AAA81r53jpUI3wLsRW6TiPCya?dl=0)
173160

161+
## Recently updates 2020.06.12
162+
163+
Add real-time segmentation model BiseNetV2 as lanenet backbone. You may modify the
164+
config/tusimple_lanenet.yaml config file to choose the front-end of lanenet model.
165+
166+
New lanenet model trainned based on BiseNetV2 can be found [here](https://www.dropbox.com/sh/0b6r0ljqi76kyg9/AADedYWO3bnx4PhK1BmbJkJKa?dl=0)
167+
168+
The new model can reach 78 fps in single image inference process.
169+
174170
## MNN Project
175171

176172
Add tools to convert lanenet tensorflow ckpt model into mnn model and deploy

tools/make_tusimple_tfrecords.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def generate_tfrecords():
2020
:return:
2121
"""
2222
producer = lanenet_data_feed_pipline.LaneNetDataProducer()
23-
producer.generate_tfrecords(step_size=1000)
23+
producer.generate_tfrecords()
2424

2525
return
2626

0 commit comments

Comments
 (0)