Skip to content

Instantly share code, notes, and snippets.

View dataplayer12's full-sized avatar

Jaiyam Sharma dataplayer12

View GitHub Profile
@dataplayer12
dataplayer12 / download_SA1B_dataset.py
Last active April 10, 2023 13:12
Download SA-1B dataset from Segment Anything paper by FAIR
import wget
import argparse
import os
import shutil
def parse_arguments():
p=argparse.ArgumentParser()
p.add_argument('--links-file', type=str, required=True, help='Path of text file downloaded from https://ai.facebook.com/datasets/segment-anything-downloads/')
p.add_argument('--start-chunk', type=int, required=False, default=0, help='Index of the starting chunk to download. Note the dataset is divided into ~1000 chunks of tar files')
p.add_argument('--end-chunk', type=int, required=False, default=-1, help='Index of the end chunk.')
@dataplayer12
dataplayer12 / CMakeLists.txt
Last active August 10, 2020 12:11
Inference on video using C++ API of retinanet
add_executable(infervideo infervideo.cpp)
target_link_libraries(infervideo PRIVATE retinanet ${OpenCV_LIBS} cuda ${CUDA_LIBRARIES})