mmsegmentation/tools
wangliujun a3aab38bb4
Update get_flops.py (#1371)
* Update get_flops.py

argparse.ArgumentParser(description='Get the FLOPs of a segmentor')

* Update get_flops.py
2022-03-14 21:02:43 +08:00
..
convert_datasets [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
model_converters [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
torchserve [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
analyze_logs.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
benchmark.py [Feature] benchmark can add work_dir and repeat times, test.py now has default work-dir (#1126) 2022-01-18 16:38:31 +08:00
browse_dataset.py [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
confusion_matrix.py [Feature] Generating and plotting confusion matrix (#1301) 2022-03-03 16:34:33 +08:00
deploy_test.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
dist_test.sh
dist_train.sh
get_flops.py Update get_flops.py (#1371) 2022-03-14 21:02:43 +08:00
onnx2tensorrt.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
print_config.py [Fix] revise --option to --options #1140 2021-12-15 15:24:08 +08:00
publish_model.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
pytorch2onnx.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
pytorch2torchscript.py [Enhancement] Add codespell pre-commit hook and fix typos (#920) 2021-10-13 06:21:17 -07:00
slurm_test.sh
slurm_train.sh
test.py [Enhance] New-style CPU training and inference. (#1251) 2022-01-28 23:03:01 +08:00
train.py [Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn (#1362) 2022-03-10 09:41:16 +08:00