keras-cv
keras-cv copied to clipboard
Published
20 hours ago •
keras-team
keras-team
Porting EfficientNets to KerasCV
@sayakpaul @LukeWood @ianjjohnson
Ported over EfficientNets (V1 and V2) from keras. Made changes wherever necessary to stick to the KerasCV code style.
/auto Closes #551
@LukeWood
The test process was killed:
Edit: The tests pass locally:
(kerascv) PS E:\GH\keras-cv> pytest keras_cv\
======================================================================================================================== test session starts ========================================================================================================================
platform win32 -- Python 3.7.13, pytest-7.1.2, pluggy-1.0.0
rootdir: E:\GH\keras-cv, configfile: setup.cfg
collected 696 items
keras_cv\version_check_test.py ... [ 0%]
keras_cv\bounding_box\converters_test.py ...............................................................................................s. [ 14%]
keras_cv\bounding_box\pad_batch_to_shape_test.py .s [ 14%]
keras_cv\bounding_box\utils_test.py .s [ 14%]
keras_cv\core\factor_sampler\constant_factor_sampler_test.py ..s [ 15%]
keras_cv\core\factor_sampler\uniform_factor_sampler_test.py ..s [ 15%]
keras_cv\keypoint\converters_test.py ................s [ 18%]
keras_cv\keypoint\utils_test.py ....s [ 18%]
keras_cv\layers\serialization_test.py ..............................s [ 23%]
keras_cv\layers\object_detection\non_max_suppression_test.py ....s. [ 24%]
keras_cv\layers\preprocessing\aug_mix_test.py ...s. [ 25%]
keras_cv\layers\preprocessing\auto_contrast_test.py .....s [ 25%]
keras_cv\layers\preprocessing\base_image_augmentation_layer_test.py ............s [ 27%]
keras_cv\layers\preprocessing\channel_shuffle_test.py ......s [ 28%]
keras_cv\layers\preprocessing\cut_mix_test.py .......s. [ 30%]
keras_cv\layers\preprocessing\equalization_test.py ........s [ 31%]
keras_cv\layers\preprocessing\fourier_mix_test.py ....s. [ 32%]
keras_cv\layers\preprocessing\grayscale_test.py ....s [ 32%]
keras_cv\layers\preprocessing\grid_mask_test.py .....s [ 33%]
keras_cv\layers\preprocessing\maybe_apply_test.py .....s.... [ 35%]
keras_cv\layers\preprocessing\mix_up_test.py ......s. [ 36%]
keras_cv\layers\preprocessing\posterization_test.py ...s... [ 37%]
keras_cv\layers\preprocessing\rand_augment_test.py ...........s... [ 39%]
keras_cv\layers\preprocessing\random_augmentation_pipeline_test.py ..........s. [ 41%]
keras_cv\layers\preprocessing\random_channel_shift_test.py ........s [ 42%]
keras_cv\layers\preprocessing\random_choice_test.py ....s [ 43%]
keras_cv\layers\preprocessing\random_color_degeneration_test.py ...s [ 43%]
keras_cv\layers\preprocessing\random_color_jitter_test.py .....s [ 44%]
keras_cv\layers\preprocessing\random_cutout_test.py .........s [ 46%]
keras_cv\layers\preprocessing\random_gaussian_blur_test.py ....s. [ 46%]
keras_cv\layers\preprocessing\random_hue_test.py .........s. [ 48%]
keras_cv\layers\preprocessing\random_jpeg_quality_test.py ...s [ 49%]
keras_cv\layers\preprocessing\random_resized_crop_test.py ........s..... [ 51%]
keras_cv\layers\preprocessing\random_rotation_test.py .....s. [ 52%]
keras_cv\layers\preprocessing\random_saturation_test.py ......s. [ 53%]
keras_cv\layers\preprocessing\random_sharpness_test.py ..s [ 53%]
keras_cv\layers\preprocessing\random_shear_test.py .........s.... [ 55%]
keras_cv\layers\preprocessing\solarization_test.py ..........s.. [ 57%]
keras_cv\layers\preprocessing\with_labels_test.py ...................................s [ 62%]
keras_cv\layers\regularization\drop_path_test.py ...s. [ 63%]
keras_cv\layers\regularization\dropblock_2d_test.py .......s. [ 64%]
keras_cv\layers\regularization\squeeze_excite_test.py ....s [ 65%]
keras_cv\layers\regularization\stochastic_depth_test.py ..s. [ 66%]
keras_cv\metrics\serialization_test.py ..s [ 66%]
keras_cv\metrics\coco\mean_average_precision_test.py ........s [ 67%]
keras_cv\metrics\coco\recall_test.py .............s.. [ 70%]
keras_cv\metrics\coco\utils_test.py ......s... [ 71%]
keras_cv\metrics\coco\numerical_tests\mean_average_precision_test.py ......s [ 72%]
keras_cv\metrics\coco\numerical_tests\recall_correctness_test.py ......s [ 73%]
keras_cv\models\models_test.py ...........................................................................................................................................................s [ 95%]
keras_cv\utils\fill_utils_test.py ......s.........s.. [ 98%]
keras_cv\utils\iou_test.py ..s [ 99%]
keras_cv\utils\preprocessing_test.py .s.... [100%]
============================================================================================================ 643 passed, 53 skipped in 642.09s (0:10:42) ============================================================================================================
(kerascv) PS E:\GH\keras-cv>
There is already an active PR for eff-v1 and eff-v2. https://github.com/keras-team/keras-cv/pull/570 https://github.com/keras-team/keras-cv/pull/587
