| import os |
| import math |
| import time |
| import numpy as np |
|
|
| class timer(): |
| def __init__(self): |
| self.acc = 0 |
| self.tic() |
|
|
| def tic(self): |
| self.t0 = time.time() |
|
|
| def toc(self, restart=False): |
| diff = time.time() - self.t0 |
| if restart: self.t0 = time.time() |
| return diff |
|
|
| def hold(self): |
| self.acc += self.toc() |
|
|
| def release(self): |
| ret = self.acc |
| self.acc = 0 |
|
|
| return ret |
|
|
| def reset(self): |
| self.acc = 0 |
|
|
| def quantize(img, rgb_range): |
| pixel_range = 255 / rgb_range |
| return img.mul(pixel_range).clamp(0, 255).round().div(pixel_range) |
|
|
