├── ascend310_infer ├── inc │ └── utils.h └── src │ ├── CMakeLists.txt │ ├── build.sh │ ├── main.cc │ └── utils.cc ├── eval.py ├── export.py ├── infer ├── README_CN.md ├── convert │ ├── aipp.cfg │ └── convert_om.sh ├── data │ └── config │ │ └── GENet.cfg ├── docker_start.sh ├── mxbase │ ├── CMakeLists.txt │ ├── build.sh │ └── src │ │ ├── GENet.cpp │ │ ├── GENet.h │ │ └── main.cpp └── sdk │ ├── GENet.pipeline │ ├── classification_task_metric.py │ ├── main.py │ └── run.sh ├── modelarts └── train_start.py ├── pytorch-GENet ├── models │ ├── blocks.py │ └── genet.py ├── train.py ├── trainer.py └── utils.py ├── readme.md ├── scripts ├── docker_start.sh ├── run_distribute_train.sh ├── run_eval.sh ├── run_infer_310.sh └── run_train.sh ├── src ├── AddOp.py ├── CrossEntropySmooth.py ├── GEBlock.py ├── GENet.py ├── config.py ├── dataset.py └── lr_generator.py ├── train.py └── unzip.py /ascend310_infer/inc/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/ascend310_infer/inc/utils.h -------------------------------------------------------------------------------- /ascend310_infer/src/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/ascend310_infer/src/CMakeLists.txt -------------------------------------------------------------------------------- /ascend310_infer/src/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/ascend310_infer/src/build.sh -------------------------------------------------------------------------------- /ascend310_infer/src/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/ascend310_infer/src/main.cc -------------------------------------------------------------------------------- /ascend310_infer/src/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/ascend310_infer/src/utils.cc -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/eval.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/export.py -------------------------------------------------------------------------------- /infer/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/README_CN.md -------------------------------------------------------------------------------- /infer/convert/aipp.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/convert/aipp.cfg -------------------------------------------------------------------------------- /infer/convert/convert_om.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/convert/convert_om.sh -------------------------------------------------------------------------------- /infer/data/config/GENet.cfg: -------------------------------------------------------------------------------- 1 | CLASS_NUM=1000 2 | SOFTMAX=false 3 | TOP_K=5 4 | -------------------------------------------------------------------------------- /infer/docker_start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/docker_start.sh -------------------------------------------------------------------------------- /infer/mxbase/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/mxbase/CMakeLists.txt -------------------------------------------------------------------------------- /infer/mxbase/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/mxbase/build.sh -------------------------------------------------------------------------------- /infer/mxbase/src/GENet.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/mxbase/src/GENet.cpp -------------------------------------------------------------------------------- /infer/mxbase/src/GENet.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/mxbase/src/GENet.h -------------------------------------------------------------------------------- /infer/mxbase/src/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/mxbase/src/main.cpp -------------------------------------------------------------------------------- /infer/sdk/GENet.pipeline: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/sdk/GENet.pipeline -------------------------------------------------------------------------------- /infer/sdk/classification_task_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/sdk/classification_task_metric.py -------------------------------------------------------------------------------- /infer/sdk/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/sdk/main.py -------------------------------------------------------------------------------- /infer/sdk/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/infer/sdk/run.sh -------------------------------------------------------------------------------- /modelarts/train_start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/modelarts/train_start.py -------------------------------------------------------------------------------- /pytorch-GENet/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/pytorch-GENet/models/blocks.py -------------------------------------------------------------------------------- /pytorch-GENet/models/genet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/pytorch-GENet/models/genet.py -------------------------------------------------------------------------------- /pytorch-GENet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/pytorch-GENet/train.py -------------------------------------------------------------------------------- /pytorch-GENet/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/pytorch-GENet/trainer.py -------------------------------------------------------------------------------- /pytorch-GENet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/pytorch-GENet/utils.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- 1 | GENet implementation of mindspore and pytorch version 2 | -------------------------------------------------------------------------------- /scripts/docker_start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/scripts/docker_start.sh -------------------------------------------------------------------------------- /scripts/run_distribute_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/scripts/run_distribute_train.sh -------------------------------------------------------------------------------- /scripts/run_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/scripts/run_eval.sh -------------------------------------------------------------------------------- /scripts/run_infer_310.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/scripts/run_infer_310.sh -------------------------------------------------------------------------------- /scripts/run_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/scripts/run_train.sh -------------------------------------------------------------------------------- /src/AddOp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/AddOp.py -------------------------------------------------------------------------------- /src/CrossEntropySmooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/CrossEntropySmooth.py -------------------------------------------------------------------------------- /src/GEBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/GEBlock.py -------------------------------------------------------------------------------- /src/GENet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/GENet.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/config.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/lr_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/src/lr_generator.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/train.py -------------------------------------------------------------------------------- /unzip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuihu1998/GENet-Res50/HEAD/unzip.py --------------------------------------------------------------------------------