暂存

.gitignore

@@ -23,10 +23,10 @@ env/
 
 uploads/*
 
-nn/dataset/toy/*
+core/dataset/toy/*
-nn/dataset/benchmark/*
+core/dataset/benchmark/*
-nn/models/*.pth
+core/models/*.pth
-nn/models/*.pt
+core/models/*.pt
 
 build/bin
 frontend/node_modules

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 = {}

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):

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):

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 core.nets.resnet import resnet
-from nn.const import label_name, input_size
+from core.const import label_name, input_size
 
 
 def test():

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 core.nets.resnet18 import resnet18
-from nn.const import label_name, input_size
+from core.const import label_name, input_size
 
 
 def test():

core/train/train_resnet.py

@@ -1,8 +1,8 @@
 import os
 import torch
-from nn.nets.resnet import resnet
+from core.nets.resnet import resnet
-from nn.dataloader.dataloader import train_dataloader
+from core.dataloader.dataloader import train_dataloader
-from nn.const import epoch, lr, batch_size
+from core.const import epoch, lr, batch_size
 
 
 def train():

core/train/train_resnet18.py

@@ -1,8 +1,8 @@
 import os
 import torch
-from nn.nets.resnet18 import resnet18
+from core.nets.resnet18 import resnet18
-from nn.dataloader.dataloader import train_dataloader
+from core.dataloader.dataloader import train_dataloader
-from nn.const import epoch, lr, batch_size
+from core.const import epoch, lr, batch_size
 
 
 def train():