├── .github └── workflows │ └── python-app.yml ├── .gitignore ├── .pylintrc ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── efficientdet ├── Det-AdvProp.md ├── README.md ├── __init__.py ├── aug │ ├── __init__.py │ ├── autoaugment.py │ ├── autoaugment_test.py │ ├── gridmask.py │ ├── gridmask_test.py │ ├── mosaic.py │ └── mosaic_test.py ├── backbone │ ├── __init__.py │ ├── backbone_factory.py │ ├── efficientnet_builder.py │ ├── efficientnet_builder_test.py │ ├── efficientnet_lite_builder.py │ ├── efficientnet_lite_builder_test.py │ ├── efficientnet_model.py │ └── efficientnet_model_test.py ├── coco_metric.py ├── coco_metric_test.py ├── dataloader.py ├── dataloader_test.py ├── dataset │ ├── README.md │ ├── __init__.py │ ├── create_coco_tfrecord.py │ ├── create_coco_tfrecord_test.py │ ├── create_pascal_tfrecord.py │ ├── create_pascal_tfrecord_test.py │ ├── inspect_tfrecords.py │ ├── label_map_util.py │ └── tfrecord_util.py ├── det_advprop_tutorial.ipynb ├── det_model_fn.py ├── det_model_fn_test.py ├── efficientdet_arch.py ├── efficientdet_arch_test.py ├── g3doc │ ├── Det-AdvProp.png │ ├── coco_ids.yaml │ ├── faq.md │ ├── flops.png │ ├── network.png │ ├── params.png │ └── street.jpg ├── hparams_config.py ├── hparams_config_test.py ├── inference.py ├── install_deps.sh ├── iou_utils.py ├── iou_utils_test.py ├── main.py ├── model_inspect.py ├── model_inspect_test.py ├── nms_np.py ├── object_detection │ ├── __init__.py │ ├── argmax_matcher.py │ ├── box_coder.py │ ├── box_list.py │ ├── faster_rcnn_box_coder.py │ ├── matcher.py │ ├── preprocessor.py │ ├── region_similarity_calculator.py │ ├── shape_utils.py │ ├── target_assigner.py │ └── tf_example_decoder.py ├── requirements.txt ├── run_tflite.py ├── tensorrt.py ├── test.sh ├── test_util.py ├── testdata │ ├── img1-d1.jpg │ └── img1.jpg ├── tf2 │ ├── README.md │ ├── __init__.py │ ├── anchors.py │ ├── efficientdet_keras.py │ ├── efficientdet_keras_test.py │ ├── eval.py │ ├── eval_tflite.py │ ├── fpn_configs.py │ ├── fpn_configs_test.py │ ├── infer.py │ ├── infer_lib.py │ ├── infer_lib_test.py │ ├── inspector.py │ ├── inspector_test.py │ ├── label_util.py │ ├── postprocess.py │ ├── postprocess_test.py │ ├── segmentation.py │ ├── tfmot.py │ ├── train.py │ ├── train_lib.py │ ├── train_lib_test.py │ ├── tutorial.ipynb │ ├── util_keras.py │ ├── util_keras_test.py │ ├── wbf.py │ └── wbf_test.py ├── tutorial.ipynb ├── utils.py ├── utils_test.py └── visualize │ ├── __init__.py │ ├── shape_utils.py │ ├── standard_fields.py │ ├── static_shape.py │ ├── vis_utils.py │ └── vis_utils_test.py ├── efficientnetv2 ├── README.md ├── autoaugment.py ├── autoaugment_test.py ├── cflags.py ├── datasets.py ├── datasets_test.py ├── effnetv2_configs.py ├── effnetv2_configs_test.py ├── effnetv2_model.py ├── effnetv2_model_test.py ├── g3doc │ ├── effnetv2-l-gpu.png │ ├── effnetv2-m-gpu.png │ ├── effnetv2-s-gpu.png │ ├── effnetv2-s-relu6-gpu.png │ ├── imagenet1k_labels.txt │ ├── imagenet21k_labels.txt │ ├── param_flops.png │ └── train_params.png ├── hparams.py ├── infer.py ├── main.py ├── main_tf2.py ├── mlir.py ├── preprocess_legacy.py ├── preprocessing.py ├── preprocessing_test.py ├── smoke_test.py ├── tfhub.ipynb ├── tutorial.ipynb ├── utils.py └── utils_test.py ├── hero ├── LICENSE ├── config_lib.py ├── core.py ├── core_test.py ├── data_lib.py ├── example_commands.sh ├── fn_lib.py ├── main.py ├── model_lib.py ├── requirements.txt ├── vb100864_openmix_v1.model └── vb32000_t5_cc.model └── lion ├── README.md ├── fig ├── ablation.png ├── alg.png ├── basic.png ├── diffusion.png ├── ft.png ├── i1k.png ├── imagen.png ├── jft-ft.png ├── jft.png ├── lit.png ├── llm.png ├── lm.png └── retrieval.png ├── lion_optax.py ├── lion_pytorch.py ├── lion_tf1.py └── lion_tf2.py /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | .DS_Store 3 | .vscode 4 | tmp 5 | .ropeproject 6 | .pyc 7 | -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/.pylintrc -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/README.md -------------------------------------------------------------------------------- /efficientdet/Det-AdvProp.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/Det-AdvProp.md -------------------------------------------------------------------------------- /efficientdet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/README.md -------------------------------------------------------------------------------- /efficientdet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/__init__.py -------------------------------------------------------------------------------- /efficientdet/aug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/__init__.py -------------------------------------------------------------------------------- /efficientdet/aug/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/autoaugment.py -------------------------------------------------------------------------------- /efficientdet/aug/autoaugment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/autoaugment_test.py -------------------------------------------------------------------------------- /efficientdet/aug/gridmask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/gridmask.py -------------------------------------------------------------------------------- /efficientdet/aug/gridmask_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/gridmask_test.py -------------------------------------------------------------------------------- /efficientdet/aug/mosaic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/mosaic.py -------------------------------------------------------------------------------- /efficientdet/aug/mosaic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/aug/mosaic_test.py -------------------------------------------------------------------------------- /efficientdet/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/__init__.py -------------------------------------------------------------------------------- /efficientdet/backbone/backbone_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/backbone_factory.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_builder.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_builder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_builder_test.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_lite_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_lite_builder.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_lite_builder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_lite_builder_test.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_model.py -------------------------------------------------------------------------------- /efficientdet/backbone/efficientnet_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/backbone/efficientnet_model_test.py -------------------------------------------------------------------------------- /efficientdet/coco_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/coco_metric.py -------------------------------------------------------------------------------- /efficientdet/coco_metric_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/coco_metric_test.py -------------------------------------------------------------------------------- /efficientdet/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataloader.py -------------------------------------------------------------------------------- /efficientdet/dataloader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataloader_test.py -------------------------------------------------------------------------------- /efficientdet/dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/README.md -------------------------------------------------------------------------------- /efficientdet/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/__init__.py -------------------------------------------------------------------------------- /efficientdet/dataset/create_coco_tfrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/create_coco_tfrecord.py -------------------------------------------------------------------------------- /efficientdet/dataset/create_coco_tfrecord_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/create_coco_tfrecord_test.py -------------------------------------------------------------------------------- /efficientdet/dataset/create_pascal_tfrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/create_pascal_tfrecord.py -------------------------------------------------------------------------------- /efficientdet/dataset/create_pascal_tfrecord_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/create_pascal_tfrecord_test.py -------------------------------------------------------------------------------- /efficientdet/dataset/inspect_tfrecords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/inspect_tfrecords.py -------------------------------------------------------------------------------- /efficientdet/dataset/label_map_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/label_map_util.py -------------------------------------------------------------------------------- /efficientdet/dataset/tfrecord_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/dataset/tfrecord_util.py -------------------------------------------------------------------------------- /efficientdet/det_advprop_tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/det_advprop_tutorial.ipynb -------------------------------------------------------------------------------- /efficientdet/det_model_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/det_model_fn.py -------------------------------------------------------------------------------- /efficientdet/det_model_fn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/det_model_fn_test.py -------------------------------------------------------------------------------- /efficientdet/efficientdet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/efficientdet_arch.py -------------------------------------------------------------------------------- /efficientdet/efficientdet_arch_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/efficientdet_arch_test.py -------------------------------------------------------------------------------- /efficientdet/g3doc/Det-AdvProp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/Det-AdvProp.png -------------------------------------------------------------------------------- /efficientdet/g3doc/coco_ids.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/coco_ids.yaml -------------------------------------------------------------------------------- /efficientdet/g3doc/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/faq.md -------------------------------------------------------------------------------- /efficientdet/g3doc/flops.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/flops.png -------------------------------------------------------------------------------- /efficientdet/g3doc/network.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/network.png -------------------------------------------------------------------------------- /efficientdet/g3doc/params.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/params.png -------------------------------------------------------------------------------- /efficientdet/g3doc/street.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/g3doc/street.jpg -------------------------------------------------------------------------------- /efficientdet/hparams_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/hparams_config.py -------------------------------------------------------------------------------- /efficientdet/hparams_config_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/hparams_config_test.py -------------------------------------------------------------------------------- /efficientdet/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/inference.py -------------------------------------------------------------------------------- /efficientdet/install_deps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/install_deps.sh -------------------------------------------------------------------------------- /efficientdet/iou_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/iou_utils.py -------------------------------------------------------------------------------- /efficientdet/iou_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/iou_utils_test.py -------------------------------------------------------------------------------- /efficientdet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/main.py -------------------------------------------------------------------------------- /efficientdet/model_inspect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/model_inspect.py -------------------------------------------------------------------------------- /efficientdet/model_inspect_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/model_inspect_test.py -------------------------------------------------------------------------------- /efficientdet/nms_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/nms_np.py -------------------------------------------------------------------------------- /efficientdet/object_detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/__init__.py -------------------------------------------------------------------------------- /efficientdet/object_detection/argmax_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/argmax_matcher.py -------------------------------------------------------------------------------- /efficientdet/object_detection/box_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/box_coder.py -------------------------------------------------------------------------------- /efficientdet/object_detection/box_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/box_list.py -------------------------------------------------------------------------------- /efficientdet/object_detection/faster_rcnn_box_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/faster_rcnn_box_coder.py -------------------------------------------------------------------------------- /efficientdet/object_detection/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/matcher.py -------------------------------------------------------------------------------- /efficientdet/object_detection/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/preprocessor.py -------------------------------------------------------------------------------- /efficientdet/object_detection/region_similarity_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/region_similarity_calculator.py -------------------------------------------------------------------------------- /efficientdet/object_detection/shape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/shape_utils.py -------------------------------------------------------------------------------- /efficientdet/object_detection/target_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/target_assigner.py -------------------------------------------------------------------------------- /efficientdet/object_detection/tf_example_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/object_detection/tf_example_decoder.py -------------------------------------------------------------------------------- /efficientdet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/requirements.txt -------------------------------------------------------------------------------- /efficientdet/run_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/run_tflite.py -------------------------------------------------------------------------------- /efficientdet/tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tensorrt.py -------------------------------------------------------------------------------- /efficientdet/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/test.sh -------------------------------------------------------------------------------- /efficientdet/test_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/test_util.py -------------------------------------------------------------------------------- /efficientdet/testdata/img1-d1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/testdata/img1-d1.jpg -------------------------------------------------------------------------------- /efficientdet/testdata/img1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/testdata/img1.jpg -------------------------------------------------------------------------------- /efficientdet/tf2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/README.md -------------------------------------------------------------------------------- /efficientdet/tf2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficientdet/tf2/anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/anchors.py -------------------------------------------------------------------------------- /efficientdet/tf2/efficientdet_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/efficientdet_keras.py -------------------------------------------------------------------------------- /efficientdet/tf2/efficientdet_keras_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/efficientdet_keras_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/eval.py -------------------------------------------------------------------------------- /efficientdet/tf2/eval_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/eval_tflite.py -------------------------------------------------------------------------------- /efficientdet/tf2/fpn_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/fpn_configs.py -------------------------------------------------------------------------------- /efficientdet/tf2/fpn_configs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/fpn_configs_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/infer.py -------------------------------------------------------------------------------- /efficientdet/tf2/infer_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/infer_lib.py -------------------------------------------------------------------------------- /efficientdet/tf2/infer_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/infer_lib_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/inspector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/inspector.py -------------------------------------------------------------------------------- /efficientdet/tf2/inspector_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/inspector_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/label_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/label_util.py -------------------------------------------------------------------------------- /efficientdet/tf2/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/postprocess.py -------------------------------------------------------------------------------- /efficientdet/tf2/postprocess_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/postprocess_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/segmentation.py -------------------------------------------------------------------------------- /efficientdet/tf2/tfmot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/tfmot.py -------------------------------------------------------------------------------- /efficientdet/tf2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/train.py -------------------------------------------------------------------------------- /efficientdet/tf2/train_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/train_lib.py -------------------------------------------------------------------------------- /efficientdet/tf2/train_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/train_lib_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/tutorial.ipynb -------------------------------------------------------------------------------- /efficientdet/tf2/util_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/util_keras.py -------------------------------------------------------------------------------- /efficientdet/tf2/util_keras_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/util_keras_test.py -------------------------------------------------------------------------------- /efficientdet/tf2/wbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/wbf.py -------------------------------------------------------------------------------- /efficientdet/tf2/wbf_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tf2/wbf_test.py -------------------------------------------------------------------------------- /efficientdet/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/tutorial.ipynb -------------------------------------------------------------------------------- /efficientdet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/utils.py -------------------------------------------------------------------------------- /efficientdet/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/utils_test.py -------------------------------------------------------------------------------- /efficientdet/visualize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/__init__.py -------------------------------------------------------------------------------- /efficientdet/visualize/shape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/shape_utils.py -------------------------------------------------------------------------------- /efficientdet/visualize/standard_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/standard_fields.py -------------------------------------------------------------------------------- /efficientdet/visualize/static_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/static_shape.py -------------------------------------------------------------------------------- /efficientdet/visualize/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/vis_utils.py -------------------------------------------------------------------------------- /efficientdet/visualize/vis_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientdet/visualize/vis_utils_test.py -------------------------------------------------------------------------------- /efficientnetv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/README.md -------------------------------------------------------------------------------- /efficientnetv2/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/autoaugment.py -------------------------------------------------------------------------------- /efficientnetv2/autoaugment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/autoaugment_test.py -------------------------------------------------------------------------------- /efficientnetv2/cflags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/cflags.py -------------------------------------------------------------------------------- /efficientnetv2/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/datasets.py -------------------------------------------------------------------------------- /efficientnetv2/datasets_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/datasets_test.py -------------------------------------------------------------------------------- /efficientnetv2/effnetv2_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/effnetv2_configs.py -------------------------------------------------------------------------------- /efficientnetv2/effnetv2_configs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/effnetv2_configs_test.py -------------------------------------------------------------------------------- /efficientnetv2/effnetv2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/effnetv2_model.py -------------------------------------------------------------------------------- /efficientnetv2/effnetv2_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/effnetv2_model_test.py -------------------------------------------------------------------------------- /efficientnetv2/g3doc/effnetv2-l-gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/effnetv2-l-gpu.png -------------------------------------------------------------------------------- /efficientnetv2/g3doc/effnetv2-m-gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/effnetv2-m-gpu.png -------------------------------------------------------------------------------- /efficientnetv2/g3doc/effnetv2-s-gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/effnetv2-s-gpu.png -------------------------------------------------------------------------------- /efficientnetv2/g3doc/effnetv2-s-relu6-gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/effnetv2-s-relu6-gpu.png -------------------------------------------------------------------------------- /efficientnetv2/g3doc/imagenet1k_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/imagenet1k_labels.txt -------------------------------------------------------------------------------- /efficientnetv2/g3doc/imagenet21k_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/imagenet21k_labels.txt -------------------------------------------------------------------------------- /efficientnetv2/g3doc/param_flops.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/param_flops.png -------------------------------------------------------------------------------- /efficientnetv2/g3doc/train_params.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/g3doc/train_params.png -------------------------------------------------------------------------------- /efficientnetv2/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/hparams.py -------------------------------------------------------------------------------- /efficientnetv2/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/infer.py -------------------------------------------------------------------------------- /efficientnetv2/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/main.py -------------------------------------------------------------------------------- /efficientnetv2/main_tf2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/main_tf2.py -------------------------------------------------------------------------------- /efficientnetv2/mlir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/mlir.py -------------------------------------------------------------------------------- /efficientnetv2/preprocess_legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/preprocess_legacy.py -------------------------------------------------------------------------------- /efficientnetv2/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/preprocessing.py -------------------------------------------------------------------------------- /efficientnetv2/preprocessing_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/preprocessing_test.py -------------------------------------------------------------------------------- /efficientnetv2/smoke_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/smoke_test.py -------------------------------------------------------------------------------- /efficientnetv2/tfhub.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/tfhub.ipynb -------------------------------------------------------------------------------- /efficientnetv2/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/tutorial.ipynb -------------------------------------------------------------------------------- /efficientnetv2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/utils.py -------------------------------------------------------------------------------- /efficientnetv2/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/efficientnetv2/utils_test.py -------------------------------------------------------------------------------- /hero/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/LICENSE -------------------------------------------------------------------------------- /hero/config_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/config_lib.py -------------------------------------------------------------------------------- /hero/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/core.py -------------------------------------------------------------------------------- /hero/core_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/core_test.py -------------------------------------------------------------------------------- /hero/data_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/data_lib.py -------------------------------------------------------------------------------- /hero/example_commands.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/example_commands.sh -------------------------------------------------------------------------------- /hero/fn_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/fn_lib.py -------------------------------------------------------------------------------- /hero/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/main.py -------------------------------------------------------------------------------- /hero/model_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/model_lib.py -------------------------------------------------------------------------------- /hero/requirements.txt: -------------------------------------------------------------------------------- 1 | jax 2 | orbax 3 | seqio 4 | einops 5 | absl-py 6 | clu 7 | t5 8 | -------------------------------------------------------------------------------- /hero/vb100864_openmix_v1.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/vb100864_openmix_v1.model -------------------------------------------------------------------------------- /hero/vb32000_t5_cc.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/hero/vb32000_t5_cc.model -------------------------------------------------------------------------------- /lion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/README.md -------------------------------------------------------------------------------- /lion/fig/ablation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/ablation.png -------------------------------------------------------------------------------- /lion/fig/alg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/alg.png -------------------------------------------------------------------------------- /lion/fig/basic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/basic.png -------------------------------------------------------------------------------- /lion/fig/diffusion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/diffusion.png -------------------------------------------------------------------------------- /lion/fig/ft.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/ft.png -------------------------------------------------------------------------------- /lion/fig/i1k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/i1k.png -------------------------------------------------------------------------------- /lion/fig/imagen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/imagen.png -------------------------------------------------------------------------------- /lion/fig/jft-ft.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/jft-ft.png -------------------------------------------------------------------------------- /lion/fig/jft.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/jft.png -------------------------------------------------------------------------------- /lion/fig/lit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/lit.png -------------------------------------------------------------------------------- /lion/fig/llm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/llm.png -------------------------------------------------------------------------------- /lion/fig/lm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/lm.png -------------------------------------------------------------------------------- /lion/fig/retrieval.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/fig/retrieval.png -------------------------------------------------------------------------------- /lion/lion_optax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/lion_optax.py -------------------------------------------------------------------------------- /lion/lion_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/lion_pytorch.py -------------------------------------------------------------------------------- /lion/lion_tf1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/lion_tf1.py -------------------------------------------------------------------------------- /lion/lion_tf2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/automl/HEAD/lion/lion_tf2.py --------------------------------------------------------------------------------