seed-project/local_train.py at master · 330233789/seed-project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from utils.operation import *
from utils.file import *

if __name__ == '__main__':
    # train(in_channels=3, out_channels=8, net_name="deeplabv3", lr=0.001, csv_path="./data_list/train.csv",
    #       load_data=copy_dataset(copy_to_local_root=r"D:\AI\project\data\baidu_lane_line\original",
    #                              source_data_path=None),
    #       batch_size=2, resize=(512, 512), crop_offset=(730, 0),
    #       epoch_begin=0, epoch_num=1,
    #       num_classes=8,
    #       save_model=save_model(local_root="./outputs/model"),
    #       load_resnet_weight=load_weight(
    #           pretrained_weights_path=r"C:\Users\zlf\.cache\torch\checkpoints\resnet101-5d3b4d8f.pth",
    #           copy_to_local_root=None),
    #       )
    train_valid(in_channels=3, out_channels=8, net_name="deeplabv3", lr=0.001,
                train_csv_path="./data_list/train1.csv",
                load_train_data=copy_dataset(copy_to_local_root=r"D:\AI\project\data\baidu_lane_line\original",
                                             source_data_path=None),
                valid_csv_path="./data_list/valid1.csv",
                load_valid_data=copy_dataset(copy_to_local_root=r"D:\AI\project\data\baidu_lane_line\original",
                                             source_data_path=None),
                batch_size=2, resize=(512, 512), crop_offset=(730, 0),
                epoch_begin=0, epoch_num=1,
                num_classes=8,
                save_model=save_model(local_root="./outputs/model"),
                loss_type=LossType.dice_loss,
                load_resnet_weight=load_weight(
                    pretrained_weights_path=r"C:\Users\zlf\.cache\torch\checkpoints\resnet101-5d3b4d8f.pth",
                    copy_to_local_root=None),
                )