{"id":21088130,"url":"https://github.com/longxingtan/tfyolo","last_synced_at":"2025-04-06T19:12:48.905Z","repository":{"id":37729911,"uuid":"325469345","full_name":"LongxingTan/tfyolo","owner":"LongxingTan","description":"tfyolo: Efficient Implementation of Yolov5 in TensorFlow","archived":false,"fork":false,"pushed_at":"2024-04-03T05:57:16.000Z","size":1390,"stargazers_count":229,"open_issues_count":15,"forks_count":71,"subscribers_count":14,"default_branch":"main","last_synced_at":"2024-10-05T16:05:34.561Z","etag":null,"topics":["object-detection","tensorflow","yolov5","you-only-look-once"],"latest_commit_sha":null,"homepage":"","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/LongxingTan.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":".github/FUNDING.yml","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},"funding":{"github":null,"patreon":null,"open_collective":null,"ko_fi":null,"tidelift":null,"community_bridge":null,"liberapay":null,"issuehunt":null,"otechie":null,"custom":"https://github.com/LongxingTan/LongxingTan/blob/main/.github/sponsor.md"}},"created_at":"2020-12-30T06:05:23.000Z","updated_at":"2024-09-25T09:26:15.000Z","dependencies_parsed_at":"2024-06-15T11:57:22.252Z","dependency_job_id":"f2838696-baf0-4a73-8865-2d0fc52904be","html_url":"https://github.com/LongxingTan/tfyolo","commit_stats":null,"previous_names":["longxingtan/tf-yolo","longxingtan/yolov5"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/LongxingTan%2Ftfyolo","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/LongxingTan%2Ftfyolo/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/LongxingTan%2Ftfyolo/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/LongxingTan%2Ftfyolo/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/LongxingTan","download_url":"https://codeload.github.com/LongxingTan/tfyolo/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247535519,"owners_count":20954576,"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","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":["object-detection","tensorflow","yolov5","you-only-look-once"],"created_at":"2024-11-19T21:15:05.424Z","updated_at":"2025-04-06T19:12:48.864Z","avatar_url":"https://github.com/LongxingTan.png","language":"Python","funding_links":["https://github.com/LongxingTan/LongxingTan/blob/main/.github/sponsor.md"],"categories":[],"sub_categories":[],"readme":"# Yolov5\n[![LICENSE](https://img.shields.io/badge/license-Anti%20996-blue.svg)](https://github.com/996icu/996.ICU/blob/master/LICENSE) \u003cbr\u003e\n\nYoloV5 implemented by TensorFlow2 , with support for training, evaluation and inference. \u003cbr\u003e\n\n\u003e **NOT perfect** project currently, but I will continue to improve this, so you might want to watch/star this repo to revisit. Any contribution is highly welcomed\u003cbr\u003e\n\n![demo](./data/sample/demo1.png)\n\n\u003c!--\n| Model | Size | AP\u003csup\u003eval\u003c/sup\u003e | AP\u003csub\u003e50\u003c/sub\u003e\u003csup\u003eval\u003c/sup\u003e | AP\u003csub\u003e75\u003c/sub\u003e\u003csup\u003eval\u003c/sup\u003e |  cfg | weights |\n| :-- | :-: | :-: | :-: | :-: | :-: | :-: | :-: | :-: | :-: | \n| YOLOV5s | 672 | 47.7% |52.6% | 61.4% | [cfg](https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/cfg/yolov4.cfg) | [weights](https://drive.google.com/file/d/137U-oLekAu-J-fe0E_seTblVxnU3tlNC/view?usp=sharing) |\n| YOLOV5m | 672 | 47.7% |52.6% | 61.4% | [cfg](https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/cfg/yolov4.cfg) | [weights](https://drive.google.com/file/d/137U-oLekAu-J-fe0E_seTblVxnU3tlNC/view?usp=sharing) |\n| YOLOV5l | 672 | 47.7% |52.6% | 61.4% | [cfg](https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/cfg/yolov4.cfg) | [weights](https://drive.google.com/file/d/137U-oLekAu-J-fe0E_seTblVxnU3tlNC/view?usp=sharing) |\n| YOLOV5x | 672 | 47.7% |52.6% | 61.4% | [cfg](https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/cfg/yolov4.cfg) | [weights](https://drive.google.com/file/d/137U-oLekAu-J-fe0E_seTblVxnU3tlNC/view?usp=sharing) |\n|  |  |  |  |  |  |  |\n--\u003e\n\n## Key Features\n- minimal Yolov5 by pure tensorflow2\n- yaml file to configure the model\n- custom data training\n- mosaic data augmentation\n- label encoding by iou or wh ratio of anchor\n- positive sample augment\n- multi-gpu training\n- detailed code comments\n- full of drawbacks with huge space to improve\n\n## Usage\n### Clone and install requirements\n```\n$ git clone git@github.com:LongxingTan/Yolov5.git\n$ cd Yolov5/\n$ pip install -r requirements.txt\n```\n\u003c!-- ### Download pretrained weights\n```\n$ cd weights/\n$ bash download_weights.sh\n``` --\u003e\n### Download VOC\n```\n$ bash data/scripts/get_voc.sh\n$ cd yolo\n$ python dataset/prepare_data.py\n```\n\n\u003c!-- ### Download COCO\n```\n$ cd data/\n$ bash get_coco_dataset.sh\n``` --\u003e\n### Train\n```\n$ python train.py\n```\n\n\n### Inference\n```\n$ python detect.py\n$ python test.py\n```\n\n### Train on custom data\nIf you want to train on custom dataset, PLEASE note the input data should like this:\n```\nimage_dir/001.jpg x_min, y_min, x_max, y_max, class_id x_min2, y_min2, x_max2, y_max2, class_id2\n```\nAnd maybe new anchor need to be created, don't forget to change the nc(number classes) in yolo-yaml.\n```\n$ python dataset/create_anchor.py\n```\n## References and Further Reading\n- [yolov5](https://github.com/ultralytics/yolov5)\n- [PyTorch_YOLOv4](https://github.com/WongKinYiu/PyTorch_YOLOv4)\n- [yolov3-tf2](https://github.com/zzh8829/yolov3-tf2)\n- [tensorflow-yolov3](https://github.com/YunYang1994/tensorflow-yolov3)\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Flongxingtan%2Ftfyolo","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Flongxingtan%2Ftfyolo","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Flongxingtan%2Ftfyolo/lists"}