charlie880624
7716a0060f
Some checks failed
deploy / build-n-publish (push) Has been cancelled
lint / lint (push) Has been cancelled
build / build_cpu (3.7, 1.5.1, torch1.5, 0.6.1) (push) Has been cancelled
build / build_cpu (3.7, 1.6.0, torch1.6, 0.7.0) (push) Has been cancelled
build / build_cpu (3.7, 1.7.0, torch1.7, 0.8.1) (push) Has been cancelled
build / build_cpu (3.7, 1.8.0, torch1.8, 0.9.0) (push) Has been cancelled
build / build_cpu (3.7, 1.9.0, torch1.9, 0.10.0) (push) Has been cancelled
build / build_cuda101 (3.7, 1.5.1+cu101, torch1.5, 0.6.1+cu101) (push) Has been cancelled
build / build_cuda101 (3.7, 1.6.0+cu101, torch1.6, 0.7.0+cu101) (push) Has been cancelled
build / build_cuda101 (3.7, 1.7.0+cu101, torch1.7, 0.8.1+cu101) (push) Has been cancelled
build / build_cuda101 (3.7, 1.8.0+cu101, torch1.8, 0.9.0+cu101) (push) Has been cancelled
build / build_cuda102 (3.6, 1.9.0+cu102, torch1.9, 0.10.0+cu102) (push) Has been cancelled
build / build_cuda102 (3.7, 1.9.0+cu102, torch1.9, 0.10.0+cu102) (push) Has been cancelled
build / build_cuda102 (3.8, 1.9.0+cu102, torch1.9, 0.10.0+cu102) (push) Has been cancelled
build / build_cuda102 (3.9, 1.9.0+cu102, torch1.9, 0.10.0+cu102) (push) Has been cancelled
build / test_windows (windows-2022, cpu, 3.8) (push) Has been cancelled
build / test_windows (windows-2022, cu111, 3.8) (push) Has been cancelled
- Add golf1/2/4/7/8 dataset classes for semantic segmentation - Add kneron-specific configs (meconfig series, kn_stdc1_golf4class) - Organize scripts into tools/check/ and tools/kneron/ - Add kneron_preprocessing module - Update README with quick-start guide - Update .gitignore to exclude data dirs, onnx, nef outputs Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import math
|
|
|
|
def round_up_16(num):
|
|
return ((num + (16 - 1)) & ~(16 - 1))
|
|
|
|
def round_up_n(num, n):
|
|
if (num > 0):
|
|
temp = float(num) / n
|
|
return math.ceil(temp) * n
|
|
else:
|
|
return -math.ceil(float(-num) / n) * n
|
|
|
|
def cal_img_row_offset(crop_num, pad_num, start_row, out_row, orig_row):
|
|
|
|
scaled_img_row = int(out_row - (pad_num[1] + pad_num[3]))
|
|
if ((start_row - pad_num[1]) > 0):
|
|
img_str_row = int((start_row - pad_num[1]))
|
|
else:
|
|
img_str_row = 0
|
|
valid_row = int(orig_row - (crop_num[1] + crop_num[3]))
|
|
img_str_row = int(valid_row * img_str_row / scaled_img_row)
|
|
return int(img_str_row + crop_num[1])
|
|
|
|
def get_pad_num(pad_num_orig, left, up, right, bottom):
|
|
pad_num = [0]*4
|
|
for i in range(0,4):
|
|
pad_num[i] = pad_num_orig[i]
|
|
|
|
if not (left):
|
|
pad_num[0] = 0
|
|
if not (up):
|
|
pad_num[1] = 0
|
|
if not (right):
|
|
pad_num[2] = 0
|
|
if not (bottom):
|
|
pad_num[3] = 0
|
|
|
|
return pad_num
|
|
|
|
def get_byte_per_pixel(raw_fmt):
|
|
if raw_fmt.lower() in ['RGB888', 'rgb888', 'RGB', 'rgb888']:
|
|
return 4
|
|
elif raw_fmt.lower() in ['YUV', 'yuv', 'YUV422', 'yuv422']:
|
|
return 2
|
|
elif raw_fmt.lower() in ['RGB565', 'rgb565']:
|
|
return 2
|
|
elif raw_fmt.lower() in ['NIR888', 'nir888', 'NIR', 'nir']:
|
|
return 1
|
|
else:
|
|
return -1 |