SAM2.1

SAM2.1 checkpoints + training code + Demo

demo/frontend/src/common/components/options/OptionButton.tsx

@@ -0,0 +1,87 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import GradientBorder from '@/common/components/button/GradientBorder';
+import useScreenSize from '@/common/screen/useScreenSize';
+import {BLUE_PINK_FILL_BR} from '@/theme/gradientStyle';
+import type {CarbonIconType} from '@carbon/icons-react';
+import {Loading} from 'react-daisyui';
+
+type Props = {
+  variant?: 'default' | 'flat' | 'gradient';
+  title: string | React.ReactNode;
+  Icon: CarbonIconType;
+  isActive?: boolean;
+  isDisabled?: boolean;
+  loadingProps?: {
+    loading: boolean;
+    label?: string;
+  };
+  onClick: () => void;
+};
+
+export default function OptionButton({
+  variant = 'default',
+  title,
+  Icon,
+  isActive = false,
+  isDisabled = false,
+  loadingProps,
+  onClick,
+}: Props) {
+  const {isMobile} = useScreenSize();
+  const isLoading = loadingProps?.loading === true;
+
+  function handleClick() {
+    if (isDisabled) {
+      return;
+    }
+    onClick();
+  }
+
+  const ButtonBase = (
+    <div
+      onClick={handleClick}
+      className={`relative rounded-lg h-full flex items-center justify-center 
+${variant === 'default' ? 'bg-graydark-700' : ''}
+${!isDisabled && 'cursor-pointer'}
+${isDisabled ? 'text-gray-300' : ''}
+${isActive && BLUE_PINK_FILL_BR}`}>
+      <div className="flex gap-2 items-center py-4 md:py-6">
+        {isLoading ? (
+          <Loading size="md" className="mx-auto mt-1" />
+        ) : (
+          <Icon
+            size={isMobile ? 24 : 28}
+            className={`mx-auto ${isDisabled ? 'text-gray-300' : 'text-white'}`}
+          />
+        )}
+        <div className="text-base font-medium text-white">
+          {isLoading && loadingProps?.label != null
+            ? loadingProps.label
+            : title}
+        </div>
+      </div>
+    </div>
+  );
+
+  return variant === 'gradient' ? (
+    <GradientBorder rounded={false} className={'rounded-lg md:rounded-full'}>
+      {ButtonBase}
+    </GradientBorder>
+  ) : (
+    ButtonBase
+  );
+}