diff --git a/.gitignore b/.gitignore index 8b56209..cbbebc9 100644 --- a/.gitignore +++ b/.gitignore @@ -23,10 +23,10 @@ env/ uploads/* -nn/dataset/toy/* -nn/dataset/benchmark/* -nn/models/*.pth -nn/models/*.pt +core/dataset/toy/* +core/dataset/benchmark/* +core/models/*.pth +core/models/*.pt build/bin frontend/node_modules diff --git a/nn/__init__.py b/core/__init__.py similarity index 100% rename from nn/__init__.py rename to core/__init__.py diff --git a/nn/const/__init__.py b/core/const/__init__.py similarity index 100% rename from nn/const/__init__.py rename to core/const/__init__.py diff --git a/nn/dataloader/__init__.py b/core/dataloader/__init__.py similarity index 100% rename from nn/dataloader/__init__.py rename to core/dataloader/__init__.py diff --git a/nn/dataloader/dataloader.py b/core/dataloader/dataloader.py similarity index 97% rename from nn/dataloader/dataloader.py rename to core/dataloader/dataloader.py index 1a3f371..a730ff4 100644 --- a/nn/dataloader/dataloader.py +++ b/core/dataloader/dataloader.py @@ -3,7 +3,7 @@ import glob from torchvision import transforms from torch.utils.data import DataLoader, Dataset from PIL import Image -from nn.const import label_name, input_size, batch_size +from core.const import label_name, input_size, batch_size label_dict = {} diff --git a/nn/nets/resnet.py b/core/nets/resnet.py similarity index 98% rename from nn/nets/resnet.py rename to core/nets/resnet.py index 71ad3b6..ea88f0c 100644 --- a/nn/nets/resnet.py +++ b/core/nets/resnet.py @@ -1,6 +1,6 @@ import torch.nn as nn import torch.nn.functional as F -from nn.const import num_classes +from core.const import num_classes class ResBlock(nn.Module): diff --git a/nn/nets/resnet18.py b/core/nets/resnet18.py similarity index 92% rename from nn/nets/resnet18.py rename to core/nets/resnet18.py index 429c8ce..61991c3 100644 --- a/nn/nets/resnet18.py +++ b/core/nets/resnet18.py @@ -1,6 +1,6 @@ import torch.nn as nn from torchvision import models -from nn.const import num_classes +from core.const import num_classes class resnet18(nn.Module): diff --git a/nn/static/dataset_stats.csv b/core/static/dataset_stats.csv similarity index 100% rename from nn/static/dataset_stats.csv rename to core/static/dataset_stats.csv diff --git a/nn/test/test_resnet.py b/core/test/test_resnet.py similarity index 93% rename from nn/test/test_resnet.py rename to core/test/test_resnet.py index 5d70a80..098e2d5 100644 --- a/nn/test/test_resnet.py +++ b/core/test/test_resnet.py @@ -4,8 +4,8 @@ import torch from torchvision import transforms from PIL import Image import numpy as np -from nn.nets.resnet import resnet -from nn.const import label_name, input_size +from core.nets.resnet import resnet +from core.const import label_name, input_size def test(): diff --git a/nn/test/test_resnet18.py b/core/test/test_resnet18.py similarity index 94% rename from nn/test/test_resnet18.py rename to core/test/test_resnet18.py index d4890fe..7e8a4e3 100644 --- a/nn/test/test_resnet18.py +++ b/core/test/test_resnet18.py @@ -4,8 +4,8 @@ import torch from torchvision import transforms from PIL import Image import numpy as np -from nn.nets.resnet18 import resnet18 -from nn.const import label_name, input_size +from core.nets.resnet18 import resnet18 +from core.const import label_name, input_size def test(): diff --git a/nn/train/train_resnet.py b/core/train/train_resnet.py similarity index 90% rename from nn/train/train_resnet.py rename to core/train/train_resnet.py index 06cc961..053b183 100644 --- a/nn/train/train_resnet.py +++ b/core/train/train_resnet.py @@ -1,8 +1,8 @@ import os import torch -from nn.nets.resnet import resnet -from nn.dataloader.dataloader import train_dataloader -from nn.const import epoch, lr, batch_size +from core.nets.resnet import resnet +from core.dataloader.dataloader import train_dataloader +from core.const import epoch, lr, batch_size def train(): diff --git a/nn/train/train_resnet18.py b/core/train/train_resnet18.py similarity index 90% rename from nn/train/train_resnet18.py rename to core/train/train_resnet18.py index 69acb4f..3cd7e73 100644 --- a/nn/train/train_resnet18.py +++ b/core/train/train_resnet18.py @@ -1,8 +1,8 @@ import os import torch -from nn.nets.resnet18 import resnet18 -from nn.dataloader.dataloader import train_dataloader -from nn.const import epoch, lr, batch_size +from core.nets.resnet18 import resnet18 +from core.dataloader.dataloader import train_dataloader +from core.const import epoch, lr, batch_size def train(): diff --git a/nn/__pycache__/const.cpython-310.pyc b/nn/__pycache__/const.cpython-310.pyc deleted file mode 100644 index 47ddbdc..0000000 Binary files a/nn/__pycache__/const.cpython-310.pyc and /dev/null differ diff --git a/nn/__pycache__/dataloader.cpython-310.pyc b/nn/__pycache__/dataloader.cpython-310.pyc deleted file mode 100644 index 3299830..0000000 Binary files a/nn/__pycache__/dataloader.cpython-310.pyc and /dev/null differ diff --git a/nn/__pycache__/dataloader.cpython-313.pyc b/nn/__pycache__/dataloader.cpython-313.pyc deleted file mode 100644 index c55cbec..0000000 Binary files a/nn/__pycache__/dataloader.cpython-313.pyc and /dev/null differ diff --git a/nn/__pycache__/resnet.cpython-310.pyc b/nn/__pycache__/resnet.cpython-310.pyc deleted file mode 100644 index 9bde1bc..0000000 Binary files a/nn/__pycache__/resnet.cpython-310.pyc and /dev/null differ diff --git a/nn/__pycache__/resnet.cpython-313.pyc b/nn/__pycache__/resnet.cpython-313.pyc deleted file mode 100644 index 3307d6b..0000000 Binary files a/nn/__pycache__/resnet.cpython-313.pyc and /dev/null differ