Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/YU1ut/imet-6th-soltuion
/ functions
Functions
49 in github.com/YU1ut/imet-6th-soltuion
⨍
Functions
49
◇
Types & classes
11
↓ 3 callers
Function
_reduce_loss
(loss)
main.py:383
↓ 3 callers
Function
load_model
(model: nn.Module, path: Path)
utils.py:28
↓ 3 callers
Function
make_loader
(df: pd.DataFrame, image_transform)
main.py:64
↓ 3 callers
Function
train
(args, model: nn.Module, criterion, *, params, train_loader, valid_loader, init_optimizer, use_cuda,
main.py:180
↓ 3 callers
Function
write_event
(log, step: int, **data)
utils.py:64
↓ 2 callers
Function
_make_mask
(argsorted, top_n: int)
main.py:375
↓ 2 callers
Function
_smooth
(ys, indices)
utils.py:165
↓ 2 callers
Function
binarize_prediction
Return matrix of 0/1 predictions, same shape as probabilities.
main.py:362
↓ 2 callers
Function
create_net
(net_cls, pretrained: bool)
models.py:16
↓ 2 callers
Function
get_ids
(root: Path)
dataset.py:83
↓ 2 callers
Function
load_image
(item, root: Path)
dataset.py:77
↓ 2 callers
Function
main
()
main.py:29
↓ 2 callers
Function
mean_df
(df: pd.DataFrame)
utils.py:24
↓ 2 callers
Function
predict
(model, root: Path, df: pd.DataFrame, out_path: Path, batch_size: int, tta: int, workers: int, use
main.py:154
↓ 2 callers
Function
validation
( model: nn.Module, criterion, valid_loader, use_cuda, )
main.py:306
↓ 1 callers
Method
__init__
(self, num_classes, pretrained=False, net_cls=M.resnet50, dropout=False)
models.py:28
↓ 1 callers
Method
__init__
(self, root: Path, df: pd.DataFrame, image_transform: Callable, tta: int)
dataset.py:43
↓ 1 callers
Method
_get_item
(self, i)
utils.py:55
↓ 1 callers
Function
cosine_rampdown
Cosine rampdown from https://arxiv.org/abs/1608.03983
main.py:356
↓ 1 callers
Function
get_score
(y_pred)
main.py:324
↓ 1 callers
Function
load_transform_image
( item, root: Path, image_transform: Callable, debug: bool = False)
dataset.py:59
↓ 1 callers
Function
main
()
make_folds.py:32
↓ 1 callers
Function
make_folds
(n_folds: int, seed: int)
make_folds.py:11
↓ 1 callers
Function
train_load_transform_image
( item, root: Path, image_transform: Callable, debug: bool = False)
dataset.py:68
Method
__call__
(self, img)
transforms.py:28
Method
__call__
(self, img)
transforms.py:62
Method
__getitem__
(self, idx: int)
dataset.py:32
Method
__getitem__
(self, idx)
dataset.py:53
Method
__init__
(self, size, interpolation=Image.BILINEAR, min_aspect=4/5, max_aspect=5/4, m
transforms.py:18
Method
__init__
(self, EPSILON = 0.5, sl = 0.02, sh = 0.4, r1 = 0.3, mean=[0.4914, 0.4822, 0.4465])
transforms.py:55
Method
__init__
(self, num_classes, pretrained=False, net_cls=M.densenet121)
models.py:49
Method
__init__
(self, root: Path, df: pd.DataFrame, image_transform: Callable, debug: bool = True)
dataset.py:21
Method
__init__
(self, gamma=2)
main.py:388
Method
__iter__
(self)
utils.py:36
Method
__len__
(self)
dataset.py:29
Method
__len__
(self)
dataset.py:50
Method
_get_item
(self, i)
utils.py:60
Function
adjust_learning_rate
if epoch <= 1: lr = np.interp([epoch], [0, 1], [0, max_lr])[0] else:
main.py:342
Method
forward
(self, x)
models.py:12
Method
forward
(self, x)
models.py:44
Method
forward
(self, x)
models.py:60
Method
forward
(self, x)
main.py:76
Method
forward
(self, logit, target)
main.py:392
Method
fresh_params
(self)
models.py:41
Method
fresh_params
(self)
models.py:57
Function
get_classes
(item)
make_submission.py:33
Function
gmean_df
(df: pd.DataFrame)
utils.py:20
Function
main
()
make_submission.py:10
Function
plot
Use in the notebook like this:: %matplotlib inline from imet.utils import plot plot('./runs/oc2', './runs/oc1', 'loss',
utils.py:72