{"id":16388238,"url":"https://github.com/tlesort/yolo-tensorrt-gie-","last_synced_at":"2025-07-23T11:39:38.889Z","repository":{"id":78327713,"uuid":"80210712","full_name":"TLESORT/YOLO-TensorRT-GIE-","owner":"TLESORT","description":"This code is an implementation of a trained YOLO neural network used with the TensorRT framework.","archived":false,"fork":false,"pushed_at":"2017-02-22T09:44:57.000Z","size":177,"stargazers_count":88,"open_issues_count":12,"forks_count":25,"subscribers_count":7,"default_branch":"master","last_synced_at":"2025-04-01T23:07:57.228Z","etag":null,"topics":["tensorrt","yolov1"],"latest_commit_sha":null,"homepage":"","language":"C++","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/TLESORT.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":"2017-01-27T13:46:06.000Z","updated_at":"2024-08-28T08:06:59.000Z","dependencies_parsed_at":"2023-03-11T14:30:51.111Z","dependency_job_id":null,"html_url":"https://github.com/TLESORT/YOLO-TensorRT-GIE-","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/TLESORT/YOLO-TensorRT-GIE-","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TLESORT%2FYOLO-TensorRT-GIE-","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TLESORT%2FYOLO-TensorRT-GIE-/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TLESORT%2FYOLO-TensorRT-GIE-/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TLESORT%2FYOLO-TensorRT-GIE-/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/TLESORT","download_url":"https://codeload.github.com/TLESORT/YOLO-TensorRT-GIE-/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/TLESORT%2FYOLO-TensorRT-GIE-/sbom","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":266671161,"owners_count":23966129,"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-07-23T02:00:09.312Z","response_time":66,"last_error":null,"robots_txt_status":null,"robots_txt_updated_at":null,"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":["tensorrt","yolov1"],"created_at":"2024-10-11T04:28:40.110Z","updated_at":"2025-07-23T11:39:38.864Z","avatar_url":"https://github.com/TLESORT.png","language":"C++","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Project YOLO-TensorRT-GIE\nThis code is an implementation of trained YOLO neural network used with the TensorRT framework.  (YOLO : \"You Only Look Once: Unified, Real-Time Object Detection\" by Joseph Redmon, Santosh Divvala, Ross Girshick, Ali Farhadi).\n\nThere is issue with this implementation :  for now the output of the neural network isn't good and I am currently searching for the solution if you have any idea don't hesitate to create an issue.\n\nThe ouputed result for detecting a cat is : \n\n![Data example](Images/cat_detection.jpg)\u003cbr /\u003e\n*Example of bad detection for a cat*\n\nWhen the ouputed result with the very same network implemented with caffe gives : \n\n![Data example](Images/true_detection.jpg)\u003cbr /\u003e\n*True detection for a cat when the network is run with the caffe framework*\n\n\nFurthermore a different images will gives very close results. For example with a matrice of zeros the result seems to be exactly the same. \n\n![Data example](Images/zeros_detection.jpg)\u003cbr /\u003e\n*Outputed detection for a matrice of zeros*\n\n\n# UPDATE :\n\nAs pointed out by AastaLLL at https://devtalk.nvidia.com/default/topic/990426/jetson-tx1/tensorrt-yolo-inference-error/post/5087820/ the leaky relu layer was not supported by TensorRT and should be remplaced  by standard-relu+scale+eltwise to approximate it. The results with the modified prototxt (yolo_small_modified.prototxt) are the following : \n\nThe 32 bits versions of tensorRT gives similar results to caffe results with yolov1-small.\n\n![Data example](Images/cat_detection_modified_32bits.jpg)\u003cbr /\u003e\n*Example of the 32 bits detection with modified prototxt*\n\n\nBut the result of the 16 bits version of tensorRT does not gives correct detections :\n\n![Data example](Images/cat_detection_modified_16bits.jpg)\u003cbr /\u003e\n*Example of the 16 bits detection with modified prototxt*\n\n![Data example](Images/zeros_detection_modified_16bits.jpg)\u003cbr /\u003e\n*Example of the 16 bits detection with modified prototxt with a matrice of zeros*\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ftlesort%2Fyolo-tensorrt-gie-","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Ftlesort%2Fyolo-tensorrt-gie-","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ftlesort%2Fyolo-tensorrt-gie-/lists"}