Skip to content

Instantly share code, notes, and snippets.

@hu8813
hu8813 / git-clone-sub.md
Created March 1, 2024 14:20 — forked from dinhvle/git-clone-sub.md
Clone Only a Subdirectory with Git
@hu8813
hu8813 / prepare.py
Created January 19, 2024 18:33
prepare.py
import numpy as np
import cv2
import time
import onnxruntime
class YOLOv8:
def __init__(self, path, conf_thres=0.3, iou_thres=0.3):
self.conf_threshold = conf_thres
self.iou_threshold = iou_thres
@hu8813
hu8813 / Static_quant.py
Created January 14, 2024 15:05
Static_quant.py
import numpy as np
import cv2
import os
from onnxruntime.quantization import quantize_static, CalibrationMethod, CalibrationDataReader, QuantType, QuantFormat
class ImageDataReader(CalibrationDataReader):
def __init__(self, image_folder, input_shape, num_samples=None):
self.image_folder = image_folder
self.input_shape = input_shape
self.images = os.listdir(image_folder)