added mobilenetv3 from torchvision and also mobilenetv3_lite models, updated docs master
authorManu Mathew <mathew.manu@ti.com>
Wed, 28 Apr 2021 05:43:23 +0000 (11:13 +0530)
committerManu Mathew <mathew.manu@ti.com>
Wed, 28 Apr 2021 05:45:00 +0000 (11:15 +0530)
commit76b6feaaf165b69525e56908171b60e6dcc3819e
tree3eb309990628c51df91372412612b66438eb81c8
parent3c851131c06083c7ba0892d6e107e896a1f6324c
added mobilenetv3 from torchvision and also mobilenetv3_lite models, updated docs
229 files changed:
LICENSE
docs/Image_Classification.md
docs/Semantic_Segmentation.md
examples/quantize_example.py [new file with mode: 0644]
examples/write_onnx_model_example.py
modules/pytorch_jacinto_ai/engine/__init__.py
modules/pytorch_jacinto_ai/engine/engine_utils.py
modules/pytorch_jacinto_ai/engine/evaluate_pixel2pixel.py
modules/pytorch_jacinto_ai/engine/infer_classification_onnx_rt.py
modules/pytorch_jacinto_ai/engine/infer_pixel2pixel.py
modules/pytorch_jacinto_ai/engine/infer_pixel2pixel_onnx_rt.py
modules/pytorch_jacinto_ai/engine/test_classification.py
modules/pytorch_jacinto_ai/engine/test_pixel2pixel_onnx.py
modules/pytorch_jacinto_ai/engine/train_classification.py
modules/pytorch_jacinto_ai/engine/train_pixel2pixel.py
modules/pytorch_jacinto_ai/xnn/__init__.py
modules/pytorch_jacinto_ai/xnn/layers/__init__.py
modules/pytorch_jacinto_ai/xnn/layers/activation.py
modules/pytorch_jacinto_ai/xnn/layers/blocks_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/layers/common_blocks.py
modules/pytorch_jacinto_ai/xnn/layers/conv_blocks.py
modules/pytorch_jacinto_ai/xnn/layers/conv_ws_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/layers/deconv_blocks.py
modules/pytorch_jacinto_ai/xnn/layers/function.py
modules/pytorch_jacinto_ai/xnn/layers/functional.py
modules/pytorch_jacinto_ai/xnn/layers/functional_blocks.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/layers/layer_config.py
modules/pytorch_jacinto_ai/xnn/layers/model_utils.py
modules/pytorch_jacinto_ai/xnn/layers/multi_task.py
modules/pytorch_jacinto_ai/xnn/layers/normalization.py
modules/pytorch_jacinto_ai/xnn/layers/quant_ste.py
modules/pytorch_jacinto_ai/xnn/layers/resize_blocks.py
modules/pytorch_jacinto_ai/xnn/layers/rf_blocks.py
modules/pytorch_jacinto_ai/xnn/onnx/__init__.py
modules/pytorch_jacinto_ai/xnn/onnx/onnx2pytorch_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/optim/__init__.py
modules/pytorch_jacinto_ai/xnn/optim/lr_scheduler.py
modules/pytorch_jacinto_ai/xnn/quantize/__init__.py
modules/pytorch_jacinto_ai/xnn/quantize/hooked_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_base_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_calib_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_graph_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_test_module.py
modules/pytorch_jacinto_ai/xnn/quantize/quant_train_module.py
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/__init__.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_base_module.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_eagercalib_module.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_eagerdistill_module.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_eagertrain_module.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_qconfig.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_qconfig_qat.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/quantize_torch_internal/quant_torch_scriptcalib_module.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/__init__.py
modules/pytorch_jacinto_ai/xnn/utils/amp.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/attr_dict.py
modules/pytorch_jacinto_ai/xnn/utils/bn_utils.py
modules/pytorch_jacinto_ai/xnn/utils/count_flops.py
modules/pytorch_jacinto_ai/xnn/utils/data_utils.py
modules/pytorch_jacinto_ai/xnn/utils/depth_utils.py
modules/pytorch_jacinto_ai/xnn/utils/export_utils_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/function_utils.py
modules/pytorch_jacinto_ai/xnn/utils/hist_utils.py
modules/pytorch_jacinto_ai/xnn/utils/image_utils.py
modules/pytorch_jacinto_ai/xnn/utils/load_weights.py
modules/pytorch_jacinto_ai/xnn/utils/logger.py
modules/pytorch_jacinto_ai/xnn/utils/module_utils.py
modules/pytorch_jacinto_ai/xnn/utils/print_utils.py
modules/pytorch_jacinto_ai/xnn/utils/quant_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/range_estimator_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/range_utils.py
modules/pytorch_jacinto_ai/xnn/utils/tensor_utils.py
modules/pytorch_jacinto_ai/xnn/utils/tensor_utils_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xnn/utils/weights_utils.py
modules/pytorch_jacinto_ai/xreferences/classification/README.md [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/classification/train.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/classification/utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/segmentation/README.md [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/segmentation/coco_utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/segmentation/run_segmentation.sh [new file with mode: 0755]
modules/pytorch_jacinto_ai/xreferences/segmentation/train.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/segmentation/transforms.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xreferences/segmentation/utils.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/LICENSE [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/README.rst [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/__init__.py
modules/pytorch_jacinto_ai/xvision/datasets/caltech.py
modules/pytorch_jacinto_ai/xvision/datasets/celeba.py
modules/pytorch_jacinto_ai/xvision/datasets/cifar.py
modules/pytorch_jacinto_ai/xvision/datasets/cityscapes.py
modules/pytorch_jacinto_ai/xvision/datasets/classification/__init__.py
modules/pytorch_jacinto_ai/xvision/datasets/coco.py
modules/pytorch_jacinto_ai/xvision/datasets/fakedata.py
modules/pytorch_jacinto_ai/xvision/datasets/flickr.py
modules/pytorch_jacinto_ai/xvision/datasets/folder.py
modules/pytorch_jacinto_ai/xvision/datasets/hmdb51.py
modules/pytorch_jacinto_ai/xvision/datasets/imagenet.py
modules/pytorch_jacinto_ai/xvision/datasets/kinetics.py
modules/pytorch_jacinto_ai/xvision/datasets/lsun.py
modules/pytorch_jacinto_ai/xvision/datasets/mnist.py
modules/pytorch_jacinto_ai/xvision/datasets/omniglot.py
modules/pytorch_jacinto_ai/xvision/datasets/phototour.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/__init__.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/a2d2.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/ade20k.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/calculate_class_weights.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/calculate_class_weights_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/cityscapes_plus.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/coco_plus.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/dataset_utils.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/flyingchairs.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/kitti_depth.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/kitti_sceneflow.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/mpisintel.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/multi_dataset_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/segmentation.py
modules/pytorch_jacinto_ai/xvision/datasets/pixel2pixel/tiad_dataset_internal.py [new file with mode: 0755]
modules/pytorch_jacinto_ai/xvision/datasets/sbd.py
modules/pytorch_jacinto_ai/xvision/datasets/sbu.py
modules/pytorch_jacinto_ai/xvision/datasets/semeion.py
modules/pytorch_jacinto_ai/xvision/datasets/stl10.py
modules/pytorch_jacinto_ai/xvision/datasets/svhn.py
modules/pytorch_jacinto_ai/xvision/datasets/ucf101.py
modules/pytorch_jacinto_ai/xvision/datasets/usps.py
modules/pytorch_jacinto_ai/xvision/datasets/utils.py
modules/pytorch_jacinto_ai/xvision/datasets/video_utils.py
modules/pytorch_jacinto_ai/xvision/datasets/vision.py
modules/pytorch_jacinto_ai/xvision/datasets/voc.py
modules/pytorch_jacinto_ai/xvision/extension.py
modules/pytorch_jacinto_ai/xvision/io/__init__.py
modules/pytorch_jacinto_ai/xvision/io/_video_opt.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/io/image.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/io/video.py
modules/pytorch_jacinto_ai/xvision/losses/__init__.py
modules/pytorch_jacinto_ai/xvision/losses/basic_loss.py
modules/pytorch_jacinto_ai/xvision/losses/flow_loss.py
modules/pytorch_jacinto_ai/xvision/losses/interest_pt_loss.py
modules/pytorch_jacinto_ai/xvision/losses/loss_utils.py
modules/pytorch_jacinto_ai/xvision/losses/norm_loss.py
modules/pytorch_jacinto_ai/xvision/losses/scale_loss.py
modules/pytorch_jacinto_ai/xvision/losses/segmentation_loss.py
modules/pytorch_jacinto_ai/xvision/losses/segmentation_loss_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/losses/unflow_loss_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/__init__.py
modules/pytorch_jacinto_ai/xvision/models/_utils.py
modules/pytorch_jacinto_ai/xvision/models/classification/__init__.py
modules/pytorch_jacinto_ai/xvision/models/densenet.py
modules/pytorch_jacinto_ai/xvision/models/detection/_utils.py
modules/pytorch_jacinto_ai/xvision/models/detection/backbone_utils.py
modules/pytorch_jacinto_ai/xvision/models/detection/faster_rcnn.py
modules/pytorch_jacinto_ai/xvision/models/detection/generalized_rcnn.py
modules/pytorch_jacinto_ai/xvision/models/detection/image_list.py
modules/pytorch_jacinto_ai/xvision/models/detection/keypoint_rcnn.py
modules/pytorch_jacinto_ai/xvision/models/detection/mask_rcnn.py
modules/pytorch_jacinto_ai/xvision/models/detection/roi_heads.py
modules/pytorch_jacinto_ai/xvision/models/detection/rpn.py
modules/pytorch_jacinto_ai/xvision/models/detection/transform.py
modules/pytorch_jacinto_ai/xvision/models/mnasnet.py
modules/pytorch_jacinto_ai/xvision/models/mobilenetv1.py
modules/pytorch_jacinto_ai/xvision/models/mobilenetv1_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv2.py
modules/pytorch_jacinto_ai/xvision/models/mobilenetv2_densenas_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv2_ericsun_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv2_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv2_shicai_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv3.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/mobilenetv3_lite.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/multi_input_net.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/__init__.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/bifpnlite_pixel2pixel.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/deeplabv3lite.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/deeplabv3lite_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/flownet_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/fpnlite_pixel2pixel.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/pixel2pixelnet.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/pixel2pixelnet_internal.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/pixel2pixelnet_utils.py
modules/pytorch_jacinto_ai/xvision/models/pixel2pixel/unetlite_pixel2pixel.py
modules/pytorch_jacinto_ai/xvision/models/regnet.py
modules/pytorch_jacinto_ai/xvision/models/resnet.py
modules/pytorch_jacinto_ai/xvision/models/segmentation/_utils.py
modules/pytorch_jacinto_ai/xvision/models/segmentation/deeplabv3.py
modules/pytorch_jacinto_ai/xvision/models/segmentation/segmentation.py
modules/pytorch_jacinto_ai/xvision/models/shufflenetv2.py
modules/pytorch_jacinto_ai/xvision/models/utils.py
modules/pytorch_jacinto_ai/xvision/ops/__init__.py
modules/pytorch_jacinto_ai/xvision/ops/_register_onnx_ops.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/ops/_utils.py
modules/pytorch_jacinto_ai/xvision/ops/boxes.py
modules/pytorch_jacinto_ai/xvision/ops/deform_conv.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/ops/feature_pyramid_network.py
modules/pytorch_jacinto_ai/xvision/ops/misc.py
modules/pytorch_jacinto_ai/xvision/ops/new_empty_tensor.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/ops/poolers.py
modules/pytorch_jacinto_ai/xvision/ops/ps_roi_align.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/ops/ps_roi_pool.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/ops/roi_align.py
modules/pytorch_jacinto_ai/xvision/ops/roi_pool.py
modules/pytorch_jacinto_ai/xvision/transforms/__init__.py
modules/pytorch_jacinto_ai/xvision/transforms/_functional_video.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/transforms/_transforms_video.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/transforms/functional.py
modules/pytorch_jacinto_ai/xvision/transforms/functional_pil.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/transforms/functional_tensor.py [new file with mode: 0644]
modules/pytorch_jacinto_ai/xvision/transforms/image_transform_utils.py
modules/pytorch_jacinto_ai/xvision/transforms/image_transforms.py
modules/pytorch_jacinto_ai/xvision/transforms/image_transforms_xv12.py
modules/pytorch_jacinto_ai/xvision/transforms/transforms.py
modules/pytorch_jacinto_ai/xvision/utils.py
requirements.txt
requirements_conda.txt
run_classification.sh
run_depth.sh
run_onnxexport.sh [new file with mode: 0755]
run_quantization.sh
run_quantization_example.sh
run_segmentation.sh
scripts/evaluate_segmentation_main.py
scripts/infer_classification_onnx_rt_main.py
scripts/infer_segmentation_main.py
scripts/infer_segmentation_onnx_main.py
scripts/test_classification_main.py
scripts/train_classification_main.py
scripts/train_depth_main.py
scripts/train_motion_segmentation_main.py
scripts/train_pixel2pixel_multitask_main.py
scripts/train_segmentation_main.py
setup.py
setup.sh
version.py