Initial codebase (#1)
* Add project code * Logger improvements * Improvements to web demo code * added create_wlasl_landmarks_dataset.py and xtract_mediapipe_landmarks.py * Fix rotation augmentation * fixed error in docstring, and removed unnecessary replace -1 -> 0 * Readme updates * Share base notebooks * Add notebooks and unify for different datasets * requirements update * fixes * Make evaluate more deterministic * Allow training with clearml * refactor preprocessing and apply linter * Minor fixes * Minor notebook tweaks * Readme updates * Fix PR comments * Remove unneeded code * Add banner to Readme --------- Co-authored-by: Gabriel Lema <gabriel.lema@xmartlabs.com>
This commit is contained in:
13
Dockerfile
Normal file
13
Dockerfile
Normal file
@@ -0,0 +1,13 @@
|
||||
FROM pytorch/pytorch
|
||||
|
||||
WORKDIR /app
|
||||
COPY ./requirements.txt /app/
|
||||
|
||||
RUN pip install -r requirements.txt
|
||||
RUN apt-get -y update
|
||||
RUN apt-get -y install git
|
||||
RUN apt-get install ffmpeg libsm6 libxext6 -y
|
||||
|
||||
COPY . /app/
|
||||
RUN git config --global --add safe.directory /app
|
||||
CMD ./train.sh
|
||||
Reference in New Issue
Block a user