{"id":16517286,"url":"https://github.com/vietanhdev/traffic-sign-classification-uff-tensorrt","last_synced_at":"2025-10-08T02:59:36.801Z","repository":{"id":79025385,"uuid":"313039967","full_name":"vietanhdev/traffic-sign-classification-uff-tensorrt","owner":"vietanhdev","description":"Traffic sign classification training and conversion code to uff - TensorRT","archived":false,"fork":false,"pushed_at":"2021-03-14T01:33:02.000Z","size":27,"stargazers_count":6,"open_issues_count":0,"forks_count":4,"subscribers_count":2,"default_branch":"master","last_synced_at":"2025-04-08T16:38:04.028Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/vietanhdev.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2020-11-15T13:42:14.000Z","updated_at":"2024-11-29T02:15:02.000Z","dependencies_parsed_at":"2023-04-07T02:24:23.115Z","dependency_job_id":null,"html_url":"https://github.com/vietanhdev/traffic-sign-classification-uff-tensorrt","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/vietanhdev/traffic-sign-classification-uff-tensorrt","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/vietanhdev%2Ftraffic-sign-classification-uff-tensorrt","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/vietanhdev%2Ftraffic-sign-classification-uff-tensorrt/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/vietanhdev%2Ftraffic-sign-classification-uff-tensorrt/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/vietanhdev%2Ftraffic-sign-classification-uff-tensorrt/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/vietanhdev","download_url":"https://codeload.github.com/vietanhdev/traffic-sign-classification-uff-tensorrt/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/vietanhdev%2Ftraffic-sign-classification-uff-tensorrt/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":273912508,"owners_count":25189969,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","status":"online","status_checked_at":"2025-09-06T02:00:13.247Z","response_time":2576,"last_error":null,"robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":true,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-10-11T16:29:45.753Z","updated_at":"2025-10-08T02:59:31.748Z","avatar_url":"https://github.com/vietanhdev.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"### Traffic sign classification\n\nThe purpose of this project is to train a traffic sign classification model for an advanced driver-assistance system. [More about this project](https://aicurious.io/posts/adas-jetson-nano-intro-and-hardware/). The inference code was integrated in [this repository](https://github.com/vietanhdev/car-smart-cam).\n\n#### Step 1: Initialize environment\n\nCreate anaconda enviroment:\n\n```conda create --name \u003cENVIROMENT_NAME\u003e python=3.6```\n\nActivate created environment and install all requirements:\n\n```pip install requirements.txt```\n\n\n#### Step 2: Train the models\n\n- Prepare dataset like below structure:\n\n![Dataset preparation](dataset-preparation.png)\n\n- Train the model:\n\n```\npython train.py\n```\n\n#### Step 3: Convert to UFF\n\n- Modify model paths in `convert_h5_to_pb.py` and `convert_pb_to_uff.py`.\n\n- Convert `.h5` model to `.pb`, and finnally `.uff`: \n\n```\npip install requirements-h5-to-uff.txt\npython convert_h5_to_pb.py\npython convert_pb_to_uff.py\n```","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fvietanhdev%2Ftraffic-sign-classification-uff-tensorrt","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fvietanhdev%2Ftraffic-sign-classification-uff-tensorrt","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fvietanhdev%2Ftraffic-sign-classification-uff-tensorrt/lists"}