├── README.md ├── data ├── training_data_example │ ├── gt_image_binary │ │ ├── 0000.png │ │ ├── 0001.png │ │ ├── 0002.png │ │ ├── 0003.png │ │ ├── 0004.png │ │ └── 0005.png │ ├── gt_image_instance │ │ ├── 0000.png │ │ ├── 0001.png │ │ ├── 0002.png │ │ ├── 0003.png │ │ ├── 0004.png │ │ └── 0005.png │ ├── image │ │ ├── 0000.png │ │ ├── 0001.png │ │ ├── 0002.png │ │ ├── 0003.png │ │ ├── 0004.png │ │ └── 0005.png │ ├── train.txt │ └── val.txt └── tusimple_test_image │ ├── 0.jpg │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ └── 4.jpg ├── lanenet ├── ConcatVideo.py ├── GetFrame.py ├── GetFrame1.py ├── Tensor2Pic.py ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ └── config.cpython-37.pyc ├── config.py ├── dataloader │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ └── transformers.cpython-37.pyc │ ├── data_loaders.py │ ├── lmdb_data_loaders.py │ └── transformers.py ├── demo_test.py ├── model │ ├── BiseNet.py │ ├── BiseNet_v2-2021-04-21.py │ ├── BiseNet_v2-bak.py │ ├── BiseNet_v2.py │ ├── BiseNet_v2_1.py │ ├── BiseNet_v2_2.py │ ├── __init__.py │ ├── __pycache__ │ │ ├── BiseNet_v2.cpython-37.pyc │ │ ├── BiseNet_v2_2.cpython-37.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── blocks.cpython-37.pyc │ │ └── loss.cpython-37.pyc │ ├── bisenetv2.py │ ├── blocks.py │ ├── decoders.py │ ├── encoders.py │ ├── lanenet_postprocess.py │ ├── loss.py │ ├── model-old.py │ └── model.py ├── online_test.py ├── online_test_video.py ├── test.py ├── train.py └── utils │ ├── PicEnhanceUtils.py │ ├── __init__.py │ ├── average_meter.py │ ├── cli_helper.py │ ├── frame0270.jpg │ ├── frame0830.jpg │ ├── frame0974.png │ ├── postprocess.py │ ├── random.jpg │ └── random1.jpg ├── scripts ├── Convert2LMDB.py ├── MultiWeightTrain.py └── generateMGdataset_v3.py ├── setup.py └── tests ├── __init__.py ├── __pycache__ └── __init__.cpython-37.pyc └── model ├── __init__.py ├── __pycache__ ├── __init__.cpython-37.pyc └── test_loss.cpython-37-PYTEST.pyc └── test_loss.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/README.md -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0000.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0001.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0002.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0003.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0004.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_binary/0005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_binary/0005.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0000.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0001.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0002.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0003.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0004.png -------------------------------------------------------------------------------- /data/training_data_example/gt_image_instance/0005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/gt_image_instance/0005.png -------------------------------------------------------------------------------- /data/training_data_example/image/0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0000.png -------------------------------------------------------------------------------- /data/training_data_example/image/0001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0001.png -------------------------------------------------------------------------------- /data/training_data_example/image/0002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0002.png -------------------------------------------------------------------------------- /data/training_data_example/image/0003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0003.png -------------------------------------------------------------------------------- /data/training_data_example/image/0004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0004.png -------------------------------------------------------------------------------- /data/training_data_example/image/0005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/image/0005.png -------------------------------------------------------------------------------- /data/training_data_example/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/train.txt -------------------------------------------------------------------------------- /data/training_data_example/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/training_data_example/val.txt -------------------------------------------------------------------------------- /data/tusimple_test_image/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/tusimple_test_image/0.jpg -------------------------------------------------------------------------------- /data/tusimple_test_image/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/tusimple_test_image/1.jpg -------------------------------------------------------------------------------- /data/tusimple_test_image/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/tusimple_test_image/2.jpg -------------------------------------------------------------------------------- /data/tusimple_test_image/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/tusimple_test_image/3.jpg -------------------------------------------------------------------------------- /data/tusimple_test_image/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/data/tusimple_test_image/4.jpg -------------------------------------------------------------------------------- /lanenet/ConcatVideo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/ConcatVideo.py -------------------------------------------------------------------------------- /lanenet/GetFrame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/GetFrame.py -------------------------------------------------------------------------------- /lanenet/GetFrame1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/GetFrame1.py -------------------------------------------------------------------------------- /lanenet/Tensor2Pic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/Tensor2Pic.py -------------------------------------------------------------------------------- /lanenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lanenet/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/config.py -------------------------------------------------------------------------------- /lanenet/dataloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lanenet/dataloader/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/dataloader/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/dataloader/__pycache__/transformers.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/dataloader/__pycache__/transformers.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/dataloader/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/dataloader/data_loaders.py -------------------------------------------------------------------------------- /lanenet/dataloader/lmdb_data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/dataloader/lmdb_data_loaders.py -------------------------------------------------------------------------------- /lanenet/dataloader/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/dataloader/transformers.py -------------------------------------------------------------------------------- /lanenet/demo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/demo_test.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet_v2-2021-04-21.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet_v2-2021-04-21.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet_v2-bak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet_v2-bak.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet_v2.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet_v2_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet_v2_1.py -------------------------------------------------------------------------------- /lanenet/model/BiseNet_v2_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/BiseNet_v2_2.py -------------------------------------------------------------------------------- /lanenet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lanenet/model/__pycache__/BiseNet_v2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/__pycache__/BiseNet_v2.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/model/__pycache__/BiseNet_v2_2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/__pycache__/BiseNet_v2_2.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/model/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/model/__pycache__/blocks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/__pycache__/blocks.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/model/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /lanenet/model/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/bisenetv2.py -------------------------------------------------------------------------------- /lanenet/model/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/blocks.py -------------------------------------------------------------------------------- /lanenet/model/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/decoders.py -------------------------------------------------------------------------------- /lanenet/model/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/encoders.py -------------------------------------------------------------------------------- /lanenet/model/lanenet_postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/lanenet_postprocess.py -------------------------------------------------------------------------------- /lanenet/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/loss.py -------------------------------------------------------------------------------- /lanenet/model/model-old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/model-old.py -------------------------------------------------------------------------------- /lanenet/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/model/model.py -------------------------------------------------------------------------------- /lanenet/online_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/online_test.py -------------------------------------------------------------------------------- /lanenet/online_test_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/online_test_video.py -------------------------------------------------------------------------------- /lanenet/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/test.py -------------------------------------------------------------------------------- /lanenet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/train.py -------------------------------------------------------------------------------- /lanenet/utils/PicEnhanceUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/PicEnhanceUtils.py -------------------------------------------------------------------------------- /lanenet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lanenet/utils/average_meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/average_meter.py -------------------------------------------------------------------------------- /lanenet/utils/cli_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/cli_helper.py -------------------------------------------------------------------------------- /lanenet/utils/frame0270.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/frame0270.jpg -------------------------------------------------------------------------------- /lanenet/utils/frame0830.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/frame0830.jpg -------------------------------------------------------------------------------- /lanenet/utils/frame0974.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/frame0974.png -------------------------------------------------------------------------------- /lanenet/utils/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/postprocess.py -------------------------------------------------------------------------------- /lanenet/utils/random.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/random.jpg -------------------------------------------------------------------------------- /lanenet/utils/random1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/lanenet/utils/random1.jpg -------------------------------------------------------------------------------- /scripts/Convert2LMDB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/scripts/Convert2LMDB.py -------------------------------------------------------------------------------- /scripts/MultiWeightTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/scripts/MultiWeightTrain.py -------------------------------------------------------------------------------- /scripts/generateMGdataset_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/scripts/generateMGdataset_v3.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/tests/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /tests/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/model/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/tests/model/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /tests/model/__pycache__/test_loss.cpython-37-PYTEST.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/tests/model/__pycache__/test_loss.cpython-37-PYTEST.pyc -------------------------------------------------------------------------------- /tests/model/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengpengfei/pytorch-lanenet/HEAD/tests/model/test_loss.py --------------------------------------------------------------------------------