Skip to content

Instantly share code, notes, and snippets.

View priyanlc's full-sized avatar

Priyan priyanlc

  • google
View GitHub Profile
@priyanlc
priyanlc / FaceImageProcessor.py
Created April 19, 2024 06:11
Identify Images with faces with face_recognition
import os
import face_recognition
class FaceImageProcessor:
@staticmethod
def process_image(filename):
try:
image = face_recognition.load_image_file(filename)
face_locations = face_recognition.face_locations(image)
@priyanlc
priyanlc / Application.py
Created April 19, 2024 06:11
Identify Images with faces with face_recognition
import dlib
from main.common.ConfigLoader import ConfigLoader
from main.face_filter.FaceImageProcessor import FaceImageProcessor
from main.face_filter.ImageBatchManager import ImageBatchManager
from pathlib import Path
class Application:
@priyanlc
priyanlc / DeleteSmall.py
Created April 19, 2024 06:05
DeleteSmallImages
import os
from main.common.ImageRemover import ImageRemover
class DeleteSmallImages:
def __init__(self, min_size_kb, max_size_mb):
self.min_size_bytes = min_size_kb * 1024
self.max_size_bytes = max_size_mb * 1024 * 1024
@priyanlc
priyanlc / Application.py
Created April 19, 2024 06:02
DeleteSmallApplication
from main.common.ConfigLoader import ConfigLoader
from main.image_filter.DeleteSmall import DeleteSmallImages
class Application:
def __init__(self, config_path):
self.config_loader = ConfigLoader(config_path)
self.image_filter = DeleteSmallImages(self.config_loader.load_config()['min_size_kb'],
self.config_loader.load_config()['max_size_mb'])
@priyanlc
priyanlc / URLManager.py
Created April 17, 2024 07:47
URLManager for ImageDatasetBuilder
class URLManager:
def __init__(self, urls_file):
self.urls_file = urls_file
def get_urls(self):
with open(self.urls_file, 'r') as file:
return [line.strip() for line in file]
@priyanlc
priyanlc / ImageDownloader.py
Created April 17, 2024 07:46
ImageDownloader for ImageDatasetBuilder
from abc import ABC, abstractmethod
from pathlib import Path
import requests
class ImageDownloader(ABC):
@abstractmethod
def download(self, url, download_loc_dir):
pass
@priyanlc
priyanlc / Application.py
Created April 17, 2024 07:44
Application file for image downloader
from concurrent.futures import ThreadPoolExecutor
import time
from pathlib import Path
from main.image_downloader.ImageDownloader import HTTPImageDownloader
from main.image_downloader.URLManager import URLManager
from main.common.ConfigLoader import ConfigLoader
@priyanlc
priyanlc / jpg_links_sample.txt
Created April 17, 2024 07:42
Sample URL file for ImageDatasetBuilder
@priyanlc
priyanlc / jpg_links_sample.csv
Created April 17, 2024 07:37
Sample URL file for ImageDatasetBuilder
{
"urls_file": "resources/links/prod/jpg_links.csv",
"download_dir": "resources/downloads",
"num_threads": 20,
"min_size_kb": 100,
"max_size_mb": 5
}