Skip to content

Instantly share code, notes, and snippets.

# init git
cd existing_folder
git init --initial-branch=main
git remote add origin git@gitlab-XXXX.com:yanwei.liu/XXXX-XXXX.git
# git commit
git add .
git commit -m "Initial commit"
git push -u origin main
import torch
import torch.nn.functional as F
def pad_image_to_nearest_multiple(image, multiple=32):
image = torch.from_numpy(image.astype(np.float32)).unsqueeze(0)
channels, height, width = image.shape
padded_height = ((height + multiple - 1) // multiple) * multiple
padded_width = ((width + multiple - 1) // multiple) * multiple
@e96031413
e96031413 / dca_multi_asset.py
Last active May 16, 2023 03:31
DCA-MultiAsset: A Python Code for Dollar-Cost Averaging Multiple Assets
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
import datetime
import pandas as pd
from pycoingecko import CoinGeckoAPI
import yfinance as yf
def get_monthly_prices_stock(symbol, date):
today = datetime.date.today()
@e96031413
e96031413 / visualize_PASCALRAW_nef.py
Created April 12, 2023 09:05
Visualizing the PASCALRAW nef format image with OpenCV and rawpy
import cv2
import imageio
import rawpy
class RawImage:
"""
A class for working with raw image files.
Attributes:
file_path (str): The path to the raw image file.
import os
import random
from shutil import copyfile
import numpy as np
from PIL import Image
import multiprocessing as mp
class DarkFace2YOLOv5:
def __init__(self, data_dir, class_list, train_ratio=0.8, random_seed=None):
self.data_dir = data_dir
import numpy as np
import cv2
import imageio
def simplest_color_balance(img, percent):
out_channels = []
channels = cv2.split(img)
for channel in channels:
total_pixels = img.shape[0] * img.shape[1]
low_val, high_val = np.percentile(channel, [percent, 100 - percent])
import os
import xml.etree.ElementTree as ET
import csv
# Set the paths for the input and output directories
voc_path = '/home/Yanwei_Liu/Datasets/PASCALRAW/annotations/'
train_img_path = '/home/Yanwei_Liu/Datasets/PASCALRAW/images/train/'
val_img_path = '/home/Yanwei_Liu/Datasets/PASCALRAW/images/val/'
train_file_path = '/home/Yanwei_Liu/Datasets/PASCALRAW/trainval/train.txt'
(Pdb) pdb.set_trace = lambda: None # This replaces the set_trace() function!
(Pdb) continue
@e96031413
e96031413 / iqa_metric.py
Created March 21, 2023 06:15
PyTorch Implementation of IQA metric Including PSNR, SSIM, LPIPS, NIQE, LOE
"""
Implementation of IQA metrics in PyTorch, including PSNR, SSIM, LPIPS, NIQE, and LOE.
"""
import torch
import torch.nn as nn
import torchvision.models as models
import torchvision.transforms.functional as F
from torch.nn.functional import conv2d
from IQA_pytorch import SSIM, LPIPSvgg
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_absolute_error, mean_squared_error
from FinMind.data import DataLoader
dl = DataLoader()
stock_data = dl.taiwan_stock_daily(
stock_id='2330', start_date='2010-01-01', end_date='2022-12-20'