/sensory-cone-detection

Sensory data group that works with the lidar and stereocamera cone detection

Primary LanguagePython

Cone detection

Sensory data group that works with the cone detection for both stereocamera and lidar, the src directory is about the training and inference for the two yolov5 models, the sensory directory contains everything needed to let our work run on the simulator. To clone the repository with its submodules:

git clone β€”recursive https://github.com/unipi-smartapp-2021/sensory-cone-detection

After that, follow all the readmes in each directory to understand how to work with both the model training and simulator detection or read the documentation files in the respective directory.

Repository structure

πŸ“‚sensory-cone-detection
β”œβ”€β”€ πŸ“‚documentation
β”‚   β”œβ”€β”€ πŸ“„Sensory_ROS_documentaion.pdf
β”‚   └── πŸ“„Sensory_Recognition_documentation.pdf
β”œβ”€β”€ πŸ“„README.md
β”œβ”€β”€ πŸ“„requirements.txt
β”œβ”€β”€ πŸ“‚sensory # this is the directory to let our work run on the simulator
β”‚   β”œβ”€β”€ πŸ“„CMakeLists.txt
β”‚   β”œβ”€β”€ πŸ“‚launch
β”‚   β”‚   β”œβ”€β”€ πŸ“„record_sensory.launch
β”‚   β”‚   └── πŸ“„sensory_ws.launch
β”‚   β”œβ”€β”€ πŸ“„package.xml
β”‚   β”œβ”€β”€ πŸ“‚scripts
β”‚   β”‚   β”œβ”€β”€ πŸ“‚cameras
β”‚   β”‚   β”‚   └── πŸ“„image0.jpg
β”‚   β”‚   β”œβ”€β”€ πŸ“„human_interaction.py
β”‚   β”‚   β”œβ”€β”€ πŸ“„lidar.py
β”‚   β”‚   β”œβ”€β”€ πŸ“‚lidars
β”‚   β”‚   β”‚   └── πŸ“„image0.jpg
β”‚   β”‚   β”œβ”€β”€ πŸ“„output_fusion.py
β”‚   β”‚   β”œβ”€β”€ πŸ“„post_processing.py
β”‚   β”‚   β”œβ”€β”€ πŸ“„rgb_camera.py
β”‚   β”‚   β”œβ”€β”€ πŸ“„utilities.py
β”‚   β”‚   β”œβ”€β”€ πŸ“‚weights
β”‚   β”‚   β”‚   β”œβ”€β”€ πŸ“„camera_weights.pt
β”‚   β”‚   β”‚   β””── πŸ“„lidar_weights.pt
β”‚   β”‚   └── πŸ“•yolov5
β”‚   β”œβ”€β”€ πŸ“„setup.py
β”‚   └── πŸ“‚src
β”‚       └── πŸ“‚sensory
β”‚           β”œβ”€β”€ πŸ“„__init__.py
β”‚           β”œβ”€β”€ πŸ“„post_processing.py
β”‚           └── πŸ“„utilities.py
└── πŸ“‚src # this directory is about the models training and message conversion
    β”œβ”€β”€ πŸ“‚datasets
    β”‚   β”œβ”€β”€ πŸ“‚example_dataset
    β”‚   β”‚   β”œβ”€β”€ πŸ“‚images
    β”‚   β”‚   β”‚   └── πŸ“„example_image.jpg
    β”‚   β”‚   └── πŸ“‚labels
    β”‚   β”‚       └── πŸ“„example_label.txt
    β”‚   └── πŸ“„README.md
    β”œβ”€β”€ πŸ“„__init__.py
    β”œβ”€β”€ πŸ“‚lidar
    β”‚   β”œβ”€β”€ πŸ“„conedataset_lidar.yaml
    β”‚   β”œβ”€β”€ πŸ“„convert_rosbag_to_pcd.py
    β”‚   β”œβ”€β”€ πŸ“„debug_box_draw.py
    β”‚   β”œβ”€β”€ πŸ“„draw_2d_box.py
    β”‚   β”œβ”€β”€ πŸ“„flatten_pcd.py
    β”‚   β”œβ”€β”€ πŸ“„full_workflow.py
    β”‚   β”œβ”€β”€ πŸ“„img2pcd.py
    β”‚   β”œβ”€β”€ πŸ“„__init__.py
    β”‚   β”œβ”€β”€ πŸ“‚interpolation_data
    β”‚   β”‚   └── πŸ“„...
    β”‚   β”œβ”€β”€ πŸ“‚interpolation_models
    β”‚   β”‚   └── πŸ“„...
    β”‚   β”œβ”€β”€ πŸ“„lidar_cone_detection.ipynb
    β”‚   β”œβ”€β”€ πŸ“„lidar_weights.pt
    β”‚   β”œβ”€β”€ πŸ“‚mathlab_exportlabel_script
    β”‚   β”‚   β”œβ”€β”€ πŸ“„export_interpolation_label.asv
    β”‚   β”‚   β”œβ”€β”€ πŸ“„export_interpolation_label.m
    β”‚   β”‚   β”œβ”€β”€ πŸ“„export_label.asv
    β”‚   β”‚   └── πŸ“„export_label.m
    β”‚   β”œβ”€β”€ πŸ“„output_fusion.py
    β”‚   β”œβ”€β”€ πŸ“„pcd2img.py
    β”‚   β”œβ”€β”€ πŸ“‚post_process_data
    β”‚   β”‚   └── πŸ“„...
    β”‚   β”œβ”€β”€ πŸ“„post_processing.py
    β”‚   β”œβ”€β”€ πŸ“„read_matlab_labels.py
    β”‚   β”œβ”€β”€ πŸ“„README.md
    β”‚   β”œβ”€β”€ πŸ“•ros_numpy
    β”‚   β”œβ”€β”€ πŸ“‚stereo
    β”‚   β”‚   └── πŸ“„...
    β”‚   └── πŸ“„utils.py
    β”œβ”€β”€ πŸ“„manage_dataset.py
    └── πŸ“‚stereocamera
        β”œβ”€β”€ πŸ“„best.pt
        β”œβ”€β”€ πŸ“„camera_cone_detection.ipynb
        β”œβ”€β”€ πŸ“„camera_detection.jpg
        β”œβ”€β”€ πŸ“„camera_detection_labels.jpg
        β”œβ”€β”€ πŸ“„conedataset_camera.yaml
        └── πŸ“„README.md

Nice directory tree, how can i create it?

sudo apt install tree

Go into the directory you want and run tree, copy the tree where you want and customize it to your needs.