Skip to content

Instantly share code, notes, and snippets.

View rishab-sharma's full-sized avatar
:octocat:
KRAKEN

Rishab Sharma rishab-sharma

:octocat:
KRAKEN
  • ISRO - Indian Space Research Organisation
  • New Delhi
View GitHub Profile
@rishab-sharma
rishab-sharma / download_gdrive_file.sh
Created October 7, 2020 13:19
Download large Google Drive Files with Wget
wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=FILEID' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=FILEID" -O FILENAME && rm -rf /tmp/cookies.txt
@rishab-sharma
rishab-sharma / asn.py
Last active September 6, 2020 11:06
Listening to the pixels - ASN
import torch
import torch.nn as nn
import torch.nn.functional as F
# Code from Hang Zhao (@hangzhaomit)
class InnerProd(nn.Module):
def __init__(self, fc_dim):
super(InnerProd, self).__init__()
self.scale = nn.Parameter(torch.ones(fc_dim))
self.bias = nn.Parameter(torch.zeros(1))
@rishab-sharma
rishab-sharma / aan.py
Last active September 6, 2020 11:05
Listening to the pixels - AAN
import torch
import torch.nn as nn
import torch.nn.functional as F
# Code from Hang Zhao (@hangzhaomit)
class Unet(nn.Module):
def __init__(self, fc_dim=64, num_downs=5, ngf=64, use_dropout=False):
super(Unet, self).__init__()
# construct unet structure
@rishab-sharma
rishab-sharma / van.py
Last active September 6, 2020 11:02
Listening to the Pixels - VAN
import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision
from functools import partial
class DilatedResnet18MaxPool(nn.Module):
def __init__(self, fc_dim=64, conv_size=3):
super(DilatedResnet18MaxPool, self).__init__()
@rishab-sharma
rishab-sharma / optical_flow_dense.py
Last active August 13, 2020 08:46
Dense Optical Flow
import cv2
import numpy as np
from glob import glob
import requests
import os
def get_video(video_url):
r = requests.get(video_url, stream = True)
with open('./vid.mp4', 'wb') as f:
for chunk in r.iter_content(chunk_size = 1024*1024):
@rishab-sharma
rishab-sharma / gcloud_instance.sh
Created October 14, 2019 09:26
Gcloud Notebook Creator
export IMAGE_FAMILY="pytorch-latest-gpu"
export ZONE="us-west1-b"
export INSTANCE_NAME="human-seg"
export INSTANCE_TYPE="n1-standard-8"
gcloud compute instances create $INSTANCE_NAME \
--zone=$ZONE \
--image-family=$IMAGE_FAMILY \
--machine-type=$INSTANCE_TYPE \
--image-project=deeplearning-platform-release \
--maintenance-policy=TERMINATE \
# coding: utf-8
__author__ = 'rishab-sharma'
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
import keras
from keras.preprocessing import image
from keras.models import Model
from keras.layers import Dense, GlobalAveragePooling2D , Dropout
@rishab-sharma
rishab-sharma / cnn_view.py
Created June 27, 2018 08:09
Visualising CNN with Keras
'''Visualization of the filters of VGG16, via gradient ascent in input space.
This script can run on CPU in a few minutes.
Results example: http://i.imgur.com/4nj4KjN.jpg
'''
from __future__ import print_function
import numpy as np
import time
from keras.preprocessing.image import save_img
from keras.applications import vgg16