-
Clone the project
git clone https://github.com/yujun0-0/MMA-Net cd MMA-Net
-
Create a conda virtual environment and activate it
conda create -n mma-net python=3.6 -y conda activate mma-net
-
Install dependencies
pip install -r requirements.txt
-
Data preparation
To run the training and testing code, we require the following data organization format
${root}-- |--${VIL100} |----JPEGImages |----Annotations |----Json |----data |------|-----db_info.yaml |------|-----test.txt |------|-----train.txt
The
root
folder can be set inoptions.py
. -
Install
CULane evaluation tools
(Only required for evaluating mIoU)If you just want to train a model or make a demo, this tool is not necessary and you can skip this step. If you want to get the evaluation results on CULane, you should install this tool.
This tools requires OpenCV C++. Please follow here to install OpenCV C++. When you build OpenCV, remove the paths of anaconda from PATH or it will be failed.
# First you need to install OpenCV C++. # After installation, make a soft link of OpenCV include path. ln -s /usr/local/include/opencv4/opencv2 /usr/local/include/opencv2
We provide three kinds of complie pipelines to build the evaluation tool of CULane. evaluate_acc Option 1:
cd evaluation/culane make
Option 2:
cd evaluation/culane mkdir build && cd build cmake .. make mv culane_evaluator ../evaluate
For Windows user:
mkdir build-vs2017 cd build-vs2017 cmake .. -G "Visual Studio 15 2017 Win64" cmake --build . --config Release # or, open the "xxx.sln" file by Visual Studio and click build button move culane_evaluator ../evaluate