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

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

write training scripts for face pixeliser #1

bibs2091 wants to merge 22 commits into from

Conversation

@bibs2091
Copy link

@bibs2091 bibs2091 commented Jul 25, 2022

Before submitting

  • Was this discussed/approved via a Jira issue?
  • Did you make sure to update the docs?
  • Did you write any new necessary tests?
  • If you made a notable change (that affects users), did you update the CHANGELOG?

What does this PR do?

This PR contains the training code for the face pixelizer and the new deploying code.
Both the train and deploy version respects isquare standards and work on them.

bibs2091 added 22 commits June 19, 2022 16:05
@bibs2091
init data_utils.py
82474fa
@bibs2091
download dataset using hugging faces
1743a6c
@bibs2091
pad the label in case there is only one face in a picture
b9aabf6
@bibs2091
init training_utils.py
383cc43
@bibs2091
add argpasers in FaceDetectorModel
4c2c6af
@bibs2091
add albumentations transforms for the bbx
b593d8c
@bibs2091
download the training set and validation set via gdown
8f859d6
@bibs2091
remove code comments
75280df
@bibs2091
add ternaus retinaface files in src
eae2486
@bibs2091
remove label.text
59a5b3f
@bibs2091
add train_config_gpu
0e6a7ab
@bibs2091
create train dockerfile
957fee7
@bibs2091
divide the repo into train and deploy folders and init them
0c5a43c
@bibs2091
add convert to onnx feature in train
aa02471
@bibs2091
add infer from onnx function
42020df
@bibs2091
optimize and structure onnx files
6e718ea
@bibs2091
Create dockerfile and worker
dbcf282
@bibs2091
Fix typo in worker
70b820e
@bibs2091
update checkpoint_callback
99fbe74
@bibs2091
update inference_onnx arguments
9bfbb14
@bibs2091
update worker and add init arguments in train.py
42f2e67
@bibs2091
update train.py and init change log
1ad7c4e
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@bibs2091