Skip to content

An user-friendly toolchain for training v2 TensorFlow object detector models

Notifications You must be signed in to change notification settings

SeanErn/EasyTfodToolchain

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

42 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EasyTfodToolchain

An user-friendly toolchain for training v2 TensorFlow object detector models


Requirements:

  • Git
  • Ubuntu-based OS (Windows is not supported at this time. Other linux distros may work, but they need to be able to run the apt package manager)
  • A GPU with CUDA support (Nvidia GPUs are recommended)
  • Training and Validation videos

Installation:

  1. Clone this repository
git clone https://github.com/SeanErn/EasyTfodToolchain.git
  1. Enter the directory
cd EasyTfodToolchain
  1. Run the installer
python run.py

Roadmap

  • Installer script fully working * Update 11/23/23: Working besides some issues with built in logger panic crashing textual when downloading CUDA via wget, tries to install cudNN when not fully downloaded Screenshot from 2023-11-23 02-04-26
  • Install error handling (report errors to user, suggest fixes)
  • Python-wide logger
  • Project Manager
  • Find annotation software (maybe labelImg or LabelStudio) OR create from scratch D:
  • Prepare, Train, Export python scripts
  • Test python script
  • FULL USER GUIDE
  • MVP
  • Advanced purge dataset, purge logs, reinstall, etc
  • TF v1 support

About

An user-friendly toolchain for training v2 TensorFlow object detector models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published