ZhangYiqin da1da48eb6 [Enhance] Add iTPN Supports for Non-three channel image (#1735) ... * Add channel argments to mae_head When trying iTPN pretrain, it only supports images with 3 channels. One of the restrictions is from MAEHead. * Transfer other argments from iTPNHiViT to HiViT The HiViT supports specifying channels, but the iTPNHiViT class can't pass channel argments to it. This is one of the reasons that iTPNHiViT implementation only support images with 3 channels. * Update itpn.py Fix hint problem		2023-09-04 13:11:16 +08:00
..
__init__.py	[Feature] Support SparK. (#1531)	2023-06-19 11:27:50 +08:00
beitv1_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
beitv2_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
cae_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
cls_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
conformer_head.py	[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)	2023-02-23 10:07:53 +08:00
contrastive_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
deit_head.py	[Refactor] Use `out_type` to specify ViT-like backbone output. (#1408)	2023-03-09 11:02:58 +08:00
efficientformer_head.py	[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)	2023-02-23 10:07:53 +08:00
grounding_head.py	[Feature] Support multiple multi-modal algorithms and inferencers. (#1561)	2023-05-19 16:50:04 +08:00
itc_head.py	[Feature] Support multiple multi-modal algorithms and inferencers. (#1561)	2023-05-19 16:50:04 +08:00
itm_head.py	[Feature] Support multiple multi-modal algorithms and inferencers. (#1561)	2023-05-19 16:50:04 +08:00
itpn_clip_head.py	[Enhance] Add GPU Acceleration Apple silicon mac (#1699)	2023-07-26 17:51:00 +08:00
latent_heads.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
levit_head.py	…
linear_head.py	…
mae_head.py	[Enhance] Add iTPN Supports for Non-three channel image (#1735)	2023-09-04 13:11:16 +08:00
margin_head.py	[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)	2023-02-23 10:07:53 +08:00
mim_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
mixmim_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
mocov3_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
multi_label_cls_head.py	[Fix] Fix the bug in binary cross entropy loss (#1499)	2023-04-19 13:53:31 +08:00
multi_label_csra_head.py	…
multi_label_linear_head.py	…
multi_task_head.py	[Fix] Fix nested predict for multi-task prediction. (#1716)	2023-07-28 13:44:12 +08:00
seq_gen_head.py	[Feature] Support multiple multi-modal algorithms and inferencers. (#1561)	2023-05-19 16:50:04 +08:00
simmim_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
spark_head.py	[Feature] Support SparK. (#1531)	2023-06-19 11:27:50 +08:00
stacked_head.py	…
swav_head.py	[Refactor] Add selfsup algorithms. (#1389)	2023-03-06 16:53:15 +08:00
vig_head.py	…
vision_transformer_head.py	[Refactor] Use `out_type` to specify ViT-like backbone output. (#1408)	2023-03-09 11:02:58 +08:00
vqa_head.py	[Feature] Support multiple multi-modal algorithms and inferencers. (#1561)	2023-05-19 16:50:04 +08:00