mmsegmentation/tools
MengzhangLI d966f98f83
[Feature] Support ICNet (#884)
* add icnet backbone

* add icnet head

* add icnet configs

* nclass -> num_classes

* Support ICNet

* ICNet

* ICNet

* Add ICNeck

* Add ICNeck

* Add ICNeck

* Add ICNeck

* Adding unittest

* Uploading models & logs

* Uploading models & logs

* add comment

* smaller test_swin.py

* try to delete test_swin.py

* delete test_unet.py

* delete test_unet.py

* temp

* smaller test_unet.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-30 09:31:57 -07:00
..
convert_datasets support coco stuff-10k/164k (#625) 2021-09-22 20:48:08 +08:00
model_converters [Enhancement] Delete convert function and add instruction to ViT/Swin README.md (#791) 2021-08-25 15:00:41 -07:00
torchserve [Fix] fix_torchserver1.1 (#844) 2021-09-23 19:14:52 +08:00
analyze_logs.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
benchmark.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
benchmark_new.py [Feature] Support ICNet (#884) 2021-09-30 09:31:57 -07:00
browse_dataset.py [Feature] Support eval concate dataset and add tool to show dataset (#833) 2021-09-09 13:00:23 +08:00
deploy_test.py [Refactor] Support progressive test with fewer memory cost (#709) 2021-08-19 20:44:58 -07:00
dist_test.sh Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
dist_train.sh init commit 2020-07-10 02:39:01 +08:00
get_flops.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
onnx2tensorrt.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
print_config.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
publish_model.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
pytorch2onnx.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
pytorch2torchscript.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
slurm_test.sh init commit 2020-07-10 02:39:01 +08:00
slurm_train.sh init commit 2020-07-10 02:39:01 +08:00
test.py [Feature] Support eval concate dataset and add tool to show dataset (#833) 2021-09-09 13:00:23 +08:00
train.py fix load ckpt bug in swin (#928) 2021-09-30 07:50:44 -07:00