diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..449e221 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +kaokore + diff --git a/code/dataloader_pytorch.py b/code/dataloader_pytorch.py index dceea82..f35d36b 100755 --- a/code/dataloader_pytorch.py +++ b/code/dataloader_pytorch.py @@ -3,7 +3,6 @@ import csv import os from PIL import Image -import torch from torch.utils.data import Dataset diff --git a/code/keras_classification_train.py b/code/keras_classification_train.py index da985d3..46f8a6a 100755 --- a/code/keras_classification_train.py +++ b/code/keras_classification_train.py @@ -1,5 +1,4 @@ import numpy as np -import os import keras as k import lycon import pandas as pd diff --git a/download.py b/download.py index 082ff84..1b85b5e 100755 --- a/download.py +++ b/download.py @@ -1,9 +1,7 @@ #!/usr/bin/env python3 import argparse -import glob import os -import subprocess import sys from os.path import join, realpath, dirname import urllib.request diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..2c50b6b --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +pillow==7.0.* +torch==1.4.* +tqdm==4.43.* +numpy==1.18.* +pandas==1.0.* +keras==2.3.* +lycon==0.2.*