🦝

LukeAI

🦝

LukeAI / batch_sam.py

Last active May 29, 2024 07:32

How to process a dir of images with SAM and save visualisations of their masks

	#!/usr/bin/env python
	from __future__ import annotations
	import os
	from pathlib import Path
	from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor
	import cv2
	import numpy as np
	import torch
	from tqdm import tqdm

LukeAI / export_yolonas_nms.py

Last active June 19, 2023 19:28

	#!/usr/bin/env python
	import os
	from super_gradients.training import models
	from super_gradients.common.object_names import Models
	import onnx
	import torch
	import torch.nn as nn


	# CONFIG

LukeAI / infer_yolonas_nms.py

Created June 16, 2023 22:27