Skip to content

Instantly share code, notes, and snippets.

View BIGBALLON's full-sized avatar
🎯
Focusing

WILL LEE BIGBALLON

🎯
Focusing
View GitHub Profile
Package Version
----------------------------- -----------
absl-py 2.1.0
accelerate 0.30.0
addict 2.4.0
aiofiles 23.2.1
aiohttp 3.9.5
aiosignal 1.3.1
aliyun-python-sdk-core 2.15.1
aliyun-python-sdk-kms 2.16.3
@BIGBALLON
BIGBALLON / radar.ipynb
Created January 13, 2024 18:52 — forked from donglixp/radar.ipynb
Radar figure in BEiT-3
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@BIGBALLON
BIGBALLON / color_log.py
Last active October 19, 2023 10:50
colorlog for color logger.
import colorlog
import logging
# 创建日志记录器
logger = colorlog.getLogger(__name__)
# 设置日志级别和格式
logger.setLevel(logging.DEBUG)
formatter = colorlog.ColoredFormatter(
"%(log_color)s%(levelname)-8s%(reset)s %(log_color)s%(asctime)s | %(blue)s%(message)s",
@BIGBALLON
BIGBALLON / imagenet21k_ids_with_classnames.csv
Created June 20, 2023 07:51
Imagenet21k label with classnames
We can't make this file beautiful and searchable because it's too large.
n00004475,organism, being
n00005787,benthos
n00006024,heterotroph
n00006484,cell
n00007846,person, individual, someone, somebody, mortal, soul
n00015388,animal, animate_being, beast, brute, creature, fauna
n00017222,plant, flora, plant_life
n00021265,food, nutrient
n00021939,artifact, artefact
n00120010,hop
@BIGBALLON
BIGBALLON / AS.py
Created January 4, 2023 08:30
Anomaly Simulation
"""
# Anomaly Simulation for MVTecAD
# Download describable textures dataset
wget https://www.robots.ox.ac.uk/~vgg/data/dtd/download/dtd-r1.0.1.tar.gz
# Download MVTec anomaly detection dataset
wget https://www.mydrive.ch/shares/38536/3830184030e49fe74747669442f0f282/download/420938113-1629952094/mvtec_anomaly_detection.tar.xz
"""
import os
import random
@BIGBALLON
BIGBALLON / EMAE_crop.py
Last active November 4, 2022 10:48
for EMAE figures
import math
import os
import random
from itertools import product
import numpy as np
from imgaug import augmenters as iaa
from PIL import Image
IMG_SIZE = 600
import numpy as np
def voc_ap(rec, prec):
# correct AP calculation
# first append sentinel values at the end
mrec = np.concatenate(([0.], rec, [1.])) #[0. 0.0666, 0.1333, 0.4 , 0.4666, 1.]
mpre = np.concatenate(([0.], prec, [0.])) #[0. 1., 0.6666, 0.4285, 0.3043, 0.]
for i in range(mpre.size - 1, 0, -1):
mpre[i - 1] = np.maximum(mpre[i - 1], mpre[i]) #[1. 1. 0.6666 0.4285 0.3043 0. ]
@BIGBALLON
BIGBALLON / conv.py
Created March 26, 2021 13:13
ddp->single
import torch
checkpoint = torch.load("resnet18.pth.tar")
state_dict = checkpoint["state_dict"]
from collections import OrderedDict
new_state_dict = OrderedDict()
for k, v in state_dict.items():
name = k[7:] # remove 'module.' of dataparallel
@BIGBALLON
BIGBALLON / .vimrc
Created March 26, 2021 10:53
vim configuration
" BASIC SETTING
set number
set noswapfile
set autoindent
set cindent
set mouse=a
set encoding=utf-8
set cursorcolumn
set cursorline
@BIGBALLON
BIGBALLON / botnet50.log
Created March 16, 2021 16:10
botnet50 training log via distribuuu
This file has been truncated, but you can view the full file.
[2021-03-14 04:47:11] LOCAL_RANK: 0, RANK: 0
[2021-03-14 04:47:11]
CFG_DEST: config.yaml
CUDNN:
BENCHMARK: true
DETERMINISTIC: false
MODEL:
ARCH: botnet50
PRETRAINED: false
SYNCBN: false