Skip to content

write training scripts for face pixeliser #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
bibs2091 wants to merge 22 commits into
base: main
Choose a base branch
from
Open

write training scripts for face pixeliser #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
82474fa
init data_utils.py
bibs2091 Jun 19, 2022
1743a6c
download dataset using hugging faces
bibs2091 Jun 20, 2022
b9aabf6
pad the label in case there is only one face in a picture
bibs2091 Jun 20, 2022
383cc43
init training_utils.py
bibs2091 Jun 20, 2022
4c2c6af
add argpasers in FaceDetectorModel
bibs2091 Jun 20, 2022
b593d8c
add albumentations transforms for the bbx
bibs2091 Jun 21, 2022
8f859d6
download the training set and validation set via gdown
bibs2091 Jun 22, 2022
75280df
remove code comments
bibs2091 Jun 22, 2022
eae2486
add ternaus retinaface files in src
bibs2091 Jun 22, 2022
59a5b3f
remove label.text
bibs2091 Jun 22, 2022
0e6a7ab
add train_config_gpu
bibs2091 Jun 22, 2022
957fee7
create train dockerfile
bibs2091 Jul 7, 2022
0c5a43c
divide the repo into train and deploy folders and init them
bibs2091 Jul 18, 2022
aa02471
add convert to onnx feature in train
bibs2091 Jul 20, 2022
42020df
add infer from onnx function
bibs2091 Jul 20, 2022
6e718ea
optimize and structure onnx files
bibs2091 Jul 21, 2022
dbcf282
Create dockerfile and worker
bibs2091 Jul 21, 2022
70b820e
Fix typo in worker
bibs2091 Jul 21, 2022
99fbe74
update checkpoint_callback
bibs2091 Jul 22, 2022
9bfbb14
update inference_onnx arguments
bibs2091 Jul 22, 2022
42f2e67
update worker and add init arguments in train.py
bibs2091 Jul 25, 2022
1ad7c4e
update train.py and init change log
bibs2091 Jul 25, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -127,3 +127,15 @@ dmypy.json

# Pyre type checker
.pyre/

# dataset cache
WIDER_train.zip
WIDER_val.zip
WIDER_train/
WIDER_val/

lightning_logs/
.vscode/
*.ckpt
.DS_Store
*.onnx
5 changes: 2 additions & 3 deletions CHANGELOG.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
# Changelog

All notable changes to this project will be documented in this file.

The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
## Version 0.1
face pixelizer train and deploy code
12 changes: 0 additions & 12 deletions Dockerfile
View file
Open in desktop

This file was deleted.

14 changes: 0 additions & 14 deletions OG.Dockerfile
View file
Open in desktop

This file was deleted.

Loading