{"id":38480246,"url":"https://github.com/wyhwong/classifier-trains","last_synced_at":"2026-01-17T05:26:10.447Z","repository":{"id":153207445,"uuid":"591545047","full_name":"wyhwong/classifier-trains","owner":"wyhwong","description":"This repository aims to provide a primitive tool to finetune state-of-the-art models with PyTorch implementation, similar to Nvidia TAO but with more flexibility in augmentation and models.","archived":false,"fork":false,"pushed_at":"2025-12-31T16:10:12.000Z","size":4030,"stargazers_count":3,"open_issues_count":0,"forks_count":2,"subscribers_count":1,"default_branch":"main","last_synced_at":"2026-01-05T01:03:35.050Z","etag":null,"topics":["computer-vision","deep-learning","image-classifier","pytorch","pytorch-lightning"],"latest_commit_sha":null,"homepage":"https://pypi.org/project/classifier_trains/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/wyhwong.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":".github/CODEOWNERS","security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null,"notice":null,"maintainers":null,"copyright":null,"agents":null,"dco":null,"cla":null}},"created_at":"2023-01-21T03:19:07.000Z","updated_at":"2025-12-31T16:10:16.000Z","dependencies_parsed_at":"2026-01-01T02:09:24.461Z","dependency_job_id":null,"html_url":"https://github.com/wyhwong/classifier-trains","commit_stats":null,"previous_names":["wyhwong/classifier-training-pipeline"],"tags_count":18,"template":false,"template_full_name":null,"purl":"pkg:github/wyhwong/classifier-trains","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wyhwong%2Fclassifier-trains","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wyhwong%2Fclassifier-trains/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wyhwong%2Fclassifier-trains/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wyhwong%2Fclassifier-trains/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/wyhwong","download_url":"https://codeload.github.com/wyhwong/classifier-trains/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wyhwong%2Fclassifier-trains/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":28499333,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-01-17T04:31:57.058Z","status":"ssl_error","status_checked_at":"2026-01-17T04:31:45.816Z","response_time":85,"last_error":"SSL_connect returned=1 errno=0 peeraddr=140.82.121.6:443 state=error: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"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":["computer-vision","deep-learning","image-classifier","pytorch","pytorch-lightning"],"created_at":"2026-01-17T05:26:10.379Z","updated_at":"2026-01-17T05:26:10.433Z","avatar_url":"https://github.com/wyhwong.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Classifier-trains\n\n[![PyPI version](https://badge.fury.io/py/classifier-trains.svg)](https://pypi.org/project/classifier-trains)\n[![Python version](https://img.shields.io/pypi/pyversions/classifier-trains)](https://pypi.org/project/classifier-trains/)\n[![Downloads](https://img.shields.io/pepy/dt/classifier-trains)](https://github.com/wyhwong/classifier-trains)\n[![license](https://img.shields.io/badge/license-MIT-orange.svg)](https://github.com/wyhwong/classifier-trains/blob/main/LICENSE)\n[![CI](https://github.com/wyhwong/classifier-trains/actions/workflows/main.yml/badge.svg)](https://github.com/wyhwong/classifier-trains/actions/workflows/main.yml/)\n\nClassifier-trains is a package aims to provide a primitive tool to finetune state-of-the-art models with PyTorch implementation, similar to Nvidia TAO but with more flexibility in augmentation and models. Like TAO classification, all parameters are configurable in yaml file to minimize the need of modifying scripts.\n\nGoal: Get config ready, get dataset ready, no coding (hopefully :D), start PyTorch training.\n\n## Install as a package\n```bash\npip3 install classifier-trains\n```\n\n## Usage\nPlease see [README.md](./src/README.md) inside `src` folder.\n\n## UML Diagram\nPlease see [UML Diagram](./docs/README.md) for the class diagram.\n\n## After Training\n\nThe logs and checkpoints will be saved in the output directory, and logs are in tensorboard format. In tensorboard, you will be able to see the ROC curve, sample images in training, parameters like learning rate and momentum, and metrics like accuracy and loss.\n\n```bash\n# Run tensorboard\ntensorboard --logdir \u003coutput_dir\u003e\n```\n\n![ROC curve in tensorboard](./docs/preview-tensorboard-1.png)\n\n![Sample images and parameters](./docs/preview-tensorboard-2.png)\n\n## Profile Report\n\nIf you run the training with profiling, a profile report will be generated in the output directory. You can see the time spent on each function. It gives you a better understanding of the performance of the training process, and a sense of where to optimize.\n\n![Profile report](./docs/preview-profile-report.png)\n\n## Features To Be Developed\n1. Implement detector training\n\n## Author\n[@wyhwong](https://github.com/wyhwong)\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fwyhwong%2Fclassifier-trains","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fwyhwong%2Fclassifier-trains","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fwyhwong%2Fclassifier-trains/lists"}