File tree 18 files changed +18
-90
lines changed
domain_adaptation/re_identification
semi_supervised_learning/image_classification
18 files changed +18
-90
lines changed Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .optim import Adam
19
18
from torch .utils .data import DataLoader
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
from tllib .vision .models .reid .loss import CrossEntropyLossWithLabelSmooth , SoftTripletLoss
23
22
from tllib .vision .models .reid .identifier import ReIdentifier
24
23
import tllib .vision .datasets .reid as datasets
32
31
33
32
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
34
33
35
- sys .path .append ('.' )
36
- import utils
37
-
38
34
39
35
def main (args : argparse .Namespace ):
40
36
logger = CompleteLogger (args .log , args .phase )
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
20
19
from torch .utils .data import DataLoader
21
20
from sklearn .cluster import KMeans , DBSCAN
22
21
23
- sys . path . append ( '../../..' )
22
+ import utils
24
23
import tllib .vision .datasets .reid as datasets
25
24
from tllib .vision .datasets .reid .convert import convert_to_pytorch_dataset
26
25
from tllib .vision .models .reid .identifier import ReIdentifier
31
30
from tllib .utils .meter import AverageMeter , ProgressMeter
32
31
from tllib .utils .logger import CompleteLogger
33
32
34
- sys .path .append ('.' )
35
- import utils
36
-
37
33
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
38
34
39
35
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import os .path as osp
11
10
19
18
from torch .utils .data import DataLoader
20
19
from sklearn .cluster import KMeans , DBSCAN
21
20
22
- sys . path . append ( '../../..' )
21
+ import utils
23
22
import tllib .vision .datasets .reid as datasets
24
23
from tllib .vision .datasets .reid .convert import convert_to_pytorch_dataset
25
24
from tllib .vision .models .reid .identifier import ReIdentifier
32
31
from tllib .utils .meter import AverageMeter , ProgressMeter
33
32
from tllib .utils .logger import CompleteLogger
34
33
35
- sys .path .append ('.' )
36
- import utils
37
-
38
34
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
39
35
40
36
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .utils .data import DataLoader
19
18
import torch .nn .functional as F
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
from tllib .alignment .coral import CorrelationAlignmentLoss
23
22
from tllib .utils .data import ForeverDataIterator
24
23
from tllib .utils .metric import accuracy
25
24
from tllib .utils .meter import AverageMeter , ProgressMeter
26
25
from tllib .utils .logger import CompleteLogger
27
26
from tllib .utils .analysis import tsne , a_distance
28
27
29
- sys .path .append ('.' )
30
- import utils
31
-
32
28
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
33
29
34
30
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .utils .data import DataLoader
19
18
import torch .nn .functional as F
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
from tllib .utils .data import ForeverDataIterator
23
22
from tllib .utils .metric import accuracy
24
23
from tllib .utils .meter import AverageMeter , ProgressMeter
25
24
from tllib .utils .logger import CompleteLogger
26
25
from tllib .utils .analysis import tsne , a_distance
27
26
28
- sys .path .append ('.' )
29
- import utils
30
-
31
27
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
32
28
33
29
Original file line number Diff line number Diff line change 6
6
import random
7
7
import time
8
8
import warnings
9
- import sys
10
9
import argparse
11
10
import shutil
12
11
import os .path as osp
19
18
from torch .utils .data import DataLoader
20
19
import torch .nn .functional as F
21
20
22
- sys . path . append ( '../../..' )
21
+ import utils
23
22
from tllib .reweight .groupdro import AutomaticUpdateDomainWeightModule
24
23
from tllib .utils .data import ForeverDataIterator
25
24
from tllib .utils .metric import accuracy
26
25
from tllib .utils .meter import AverageMeter , ProgressMeter
27
26
from tllib .utils .logger import CompleteLogger
28
27
from tllib .utils .analysis import tsne , a_distance
29
28
30
- sys .path .append ('.' )
31
- import utils
32
-
33
29
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
34
30
35
31
Original file line number Diff line number Diff line change 6
6
import random
7
7
import time
8
8
import warnings
9
- import sys
10
9
import argparse
11
10
import shutil
12
11
import os .path as osp
20
19
import torch .nn .functional as F
21
20
import torch .autograd as autograd
22
21
23
- sys . path . append ( '../../..' )
22
+ import utils
24
23
from tllib .utils .data import ForeverDataIterator
25
24
from tllib .utils .metric import accuracy
26
25
from tllib .utils .meter import AverageMeter , ProgressMeter
27
26
from tllib .utils .logger import CompleteLogger
28
27
from tllib .utils .analysis import tsne , a_distance
29
28
30
- sys .path .append ('.' )
31
- import utils
32
-
33
29
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
34
30
35
31
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .utils .data import DataLoader
19
18
import torch .nn .functional as F
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
import tllib .normalization .mixstyle .resnet as models
23
22
from tllib .utils .data import ForeverDataIterator
24
23
from tllib .utils .metric import accuracy
25
24
from tllib .utils .meter import AverageMeter , ProgressMeter
26
25
from tllib .utils .logger import CompleteLogger
27
26
from tllib .utils .analysis import tsne , a_distance
28
27
29
- sys .path .append ('.' )
30
- import utils
31
-
32
28
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
33
29
34
30
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
19
18
import torch .nn .functional as F
20
19
import higher
21
20
22
- sys . path . append ( '../../..' )
21
+ import utils
23
22
from tllib .utils .data import ForeverDataIterator
24
23
from tllib .utils .metric import accuracy
25
24
from tllib .utils .meter import AverageMeter , ProgressMeter
26
25
from tllib .utils .logger import CompleteLogger
27
26
from tllib .utils .analysis import tsne , a_distance
28
27
29
- sys .path .append ('.' )
30
- import utils
31
-
32
28
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
33
29
34
30
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .utils .data import DataLoader
19
18
import torch .nn .functional as F
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
from tllib .utils .data import ForeverDataIterator
23
22
from tllib .utils .metric import accuracy
24
23
from tllib .utils .meter import AverageMeter , ProgressMeter
25
24
from tllib .utils .logger import CompleteLogger
26
25
from tllib .utils .analysis import tsne , a_distance
27
26
28
- sys .path .append ('.' )
29
- import utils
30
-
31
27
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
32
28
33
29
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
18
17
from torch .optim import Adam
19
18
from torch .utils .data import DataLoader
20
19
21
- sys . path . append ( '../../..' )
20
+ import utils
22
21
from tllib .vision .models .reid .loss import CrossEntropyLossWithLabelSmooth , SoftTripletLoss
23
22
from tllib .vision .models .reid .identifier import ReIdentifier
24
23
import tllib .vision .datasets .reid as datasets
30
29
from tllib .utils .meter import AverageMeter , ProgressMeter
31
30
from tllib .utils .logger import CompleteLogger
32
31
33
- sys .path .append ('.' )
34
- import utils
35
-
36
32
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
37
33
38
34
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
import os .path as osp
17
16
from torch .optim import Adam
18
17
from torch .utils .data import DataLoader
19
18
20
- sys . path . append ( '../../..' )
19
+ import utils
21
20
from tllib .normalization .mixstyle .sampler import RandomDomainMultiInstanceSampler
22
21
import tllib .normalization .mixstyle .resnet as models
23
22
from tllib .vision .models .reid .identifier import ReIdentifier
32
31
from tllib .utils .meter import AverageMeter , ProgressMeter
33
32
from tllib .utils .logger import CompleteLogger
34
33
35
- sys .path .append ('.' )
36
- import utils
37
-
38
34
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
39
35
40
36
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
17
16
from torch .optim .lr_scheduler import LambdaLR
18
17
from torch .utils .data import DataLoader , ConcatDataset
19
18
20
- sys . path . append ( '../../..' )
19
+ import utils
21
20
from tllib .vision .transforms import MultipleApply
22
21
from tllib .utils .metric import accuracy
23
22
from tllib .utils .meter import AverageMeter , ProgressMeter
24
23
from tllib .utils .data import ForeverDataIterator
25
24
from tllib .utils .logger import CompleteLogger
26
25
27
- sys .path .append ('.' )
28
- import utils
29
-
30
26
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
31
27
32
28
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
17
16
from torch .optim .lr_scheduler import LambdaLR
18
17
from torch .utils .data import DataLoader
19
18
20
- sys . path . append ( '../../..' )
19
+ import utils
21
20
from tllib .self_training .pseudo_label import ConfidenceRegularizedSelfTrainingLoss
22
21
from tllib .vision .transforms import MultipleApply
23
22
from tllib .utils .metric import accuracy
24
23
from tllib .utils .meter import AverageMeter , ProgressMeter
25
24
from tllib .utils .data import ForeverDataIterator
26
25
from tllib .utils .logger import CompleteLogger
27
26
28
- sys .path .append ('.' )
29
- import utils
30
-
31
27
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
32
28
33
29
Original file line number Diff line number Diff line change 5
5
import random
6
6
import time
7
7
import warnings
8
- import sys
9
8
import argparse
10
9
import shutil
11
10
17
16
from torch .optim .lr_scheduler import LambdaLR
18
17
from torch .utils .data import DataLoader
19
18
20
- sys . path . append ( '../../..' )
19
+ import utils
21
20
from tllib .self_training .pi_model import sigmoid_warm_up , L2ConsistencyLoss
22
21
from tllib .self_training .mean_teacher import update_bn , EMATeacher
23
22
from tllib .vision .transforms import MultipleApply
26
25
from tllib .utils .data import ForeverDataIterator
27
26
from tllib .utils .logger import CompleteLogger
28
27
29
- sys .path .append ('.' )
30
- import utils
31
-
32
28
device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
33
29
34
30
You can’t perform that action at this time.
0 commit comments