Skip to content

Commit

Permalink
updates
Browse files Browse the repository at this point in the history
  • Loading branch information
pprp committed Jan 30, 2020
1 parent 2c4d6a4 commit da93ae0
Show file tree
Hide file tree
Showing 291 changed files with 1,293 additions and 29 deletions.
1 change: 1 addition & 0 deletions SmallObjectAugmentation/aug.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,6 +115,7 @@ def copysmallobjects2(image_dir, label_dir, save_base_dir, small_img_dir):
for small_img_dirs in small_img_dir:
image_bbox = cv2.imread(small_img_dirs)
#roi = image_bbox
# TODO
roi = suo_fang(image_bbox,area_max=3000,area_min=1500)

new_bboxes = random_add_patches2(roi.shape, rescale_labels, image.shape, paste_number=1, iou_thresh=0)
Expand Down
53 changes: 53 additions & 0 deletions SmallObjectAugmentation/crop_img.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
import os
import shutil
from os.path import join
import cv2
import glob

root_dir = "./dpjsamplecrop"
save_dir = "./dpjcrops"

jpg_list = glob.glob(root_dir + "/*.jpg")


fo = open("dpj_small.txt", "w")

for jpg_path in jpg_list:
# jpg_path = jpg_list[3]
txt_path = jpg_path.replace("jpg", "txt")
jpg_name = os.path.basename(jpg_path)

f = open(txt_path, "r")

img = cv2.imread(jpg_path)

height, width, channel = img.shape

file_contents = f.readlines()

for num, file_content in enumerate(file_contents):
print(num)
clss, xc, yc, w, h = file_content.split()
xc, yc, w, h = float(xc), float(yc), float(w), float(h)

xc *= width
yc *= height
w *= width
h *= height

half_w, half_h = w // 2, h // 2

x1, y1 = int(xc - half_w), int(yc - half_h)
x2, y2 = int(xc + half_w), int(yc + half_h)

crop_img = img[y1:y2, x1:x2]

new_jpg_name = jpg_name.split('.')[0] + "_crop_" + str(num) + ".jpg"
cv2.imwrite(os.path.join(save_dir, new_jpg_name), crop_img)
# cv2.imshow("croped",crop_img)
# cv2.waitKey(0)
fo.write(os.path.join(save_dir, new_jpg_name)+"\n")

f.close()

fo.close()
18 changes: 10 additions & 8 deletions SmallObjectAugmentation/demo.py
Original file line number Diff line number Diff line change
@@ -1,29 +1,31 @@
import aug as am
import Helpers as hp
from util import *
import os
import random
from os.path import join

from tqdm import tqdm
import random

import aug as am
import Helpers as hp
from util import *

base_dir = os.getcwd()

save_base_dir = join(base_dir, 'save')
save_base_dir = join(base_dir, 'dpjsave')

check_dir(save_base_dir)

imgs_dir = [f.strip() for f in open(join(base_dir, 'train.txt')).readlines()]
imgs_dir = [f.strip() for f in open(join(base_dir, 'dpj_train.txt')).readlines()]
labels_dir = hp.replace_labels(imgs_dir)

small_imgs_dir = [f.strip() for f in open(join(base_dir, 'small.txt')).readlines()]
small_imgs_dir = [f.strip() for f in open(join(base_dir, 'dpj_small.txt')).readlines()]
random.shuffle(small_imgs_dir)

for image_dir, label_dir in tqdm(zip(imgs_dir, labels_dir)):
small_img = []
for x in range(8):
if small_imgs_dir == []:
#exit()
small_imgs_dir = [f.strip() for f in open(join(base_dir,'small.txt')).readlines()]
small_imgs_dir = [f.strip() for f in open(join(base_dir,'dpj_small.txt')).readlines()]
random.shuffle(small_imgs_dir)
small_img.append(small_imgs_dir.pop())
am.copysmallobjects2(image_dir, label_dir, save_base_dir,small_img)
193 changes: 193 additions & 0 deletions SmallObjectAugmentation/dpj_small.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,193 @@
./dpjcrops\1070_crop_0.jpg
./dpjcrops\109_crop_0.jpg
./dpjcrops\1169_crop_0.jpg
./dpjcrops\117_crop_0.jpg
./dpjcrops\1262_crop_0.jpg
./dpjcrops\1322_crop_0.jpg
./dpjcrops\1392_crop_0.jpg
./dpjcrops\1394_crop_0.jpg
./dpjcrops\1445_crop_0.jpg
./dpjcrops\1507_crop_0.jpg
./dpjcrops\152_crop_0.jpg
./dpjcrops\158_crop_0.jpg
./dpjcrops\1724_crop_0.jpg
./dpjcrops\1728_crop_0.jpg
./dpjcrops\1757_crop_0.jpg
./dpjcrops\1783_crop_0.jpg
./dpjcrops\1790_crop_0.jpg
./dpjcrops\1817_crop_0.jpg
./dpjcrops\1820_crop_0.jpg
./dpjcrops\1822_crop_0.jpg
./dpjcrops\1865_crop_0.jpg
./dpjcrops\1890_crop_0.jpg
./dpjcrops\1890_crop_1.jpg
./dpjcrops\1896_crop_0.jpg
./dpjcrops\1896_crop_1.jpg
./dpjcrops\1919_crop_0.jpg
./dpjcrops\1919_crop_1.jpg
./dpjcrops\192_crop_0.jpg
./dpjcrops\1920_crop_0.jpg
./dpjcrops\1920_crop_1.jpg
./dpjcrops\1939_crop_0.jpg
./dpjcrops\1956_crop_0.jpg
./dpjcrops\1965_crop_0.jpg
./dpjcrops\1976_crop_0.jpg
./dpjcrops\2007_crop_0.jpg
./dpjcrops\2049_crop_0.jpg
./dpjcrops\2050_crop_0.jpg
./dpjcrops\2131_crop_0.jpg
./dpjcrops\2146_crop_0.jpg
./dpjcrops\2150_crop_0.jpg
./dpjcrops\2174_crop_0.jpg
./dpjcrops\2185_crop_0.jpg
./dpjcrops\2203_crop_0.jpg
./dpjcrops\2216_crop_0.jpg
./dpjcrops\2223_crop_0.jpg
./dpjcrops\2253_crop_0.jpg
./dpjcrops\2254_crop_0.jpg
./dpjcrops\2276_crop_0.jpg
./dpjcrops\2277_crop_0.jpg
./dpjcrops\2277_crop_1.jpg
./dpjcrops\2290_crop_0.jpg
./dpjcrops\2305_crop_0.jpg
./dpjcrops\2310_crop_0.jpg
./dpjcrops\2330_crop_0.jpg
./dpjcrops\2361_crop_0.jpg
./dpjcrops\2501_crop_0.jpg
./dpjcrops\254_crop_0.jpg
./dpjcrops\258_crop_0.jpg
./dpjcrops\260_crop_0.jpg
./dpjcrops\2659_crop_0.jpg
./dpjcrops\274_crop_0.jpg
./dpjcrops\276_crop_0.jpg
./dpjcrops\288_crop_0.jpg
./dpjcrops\293_crop_0.jpg
./dpjcrops\3017_crop_0.jpg
./dpjcrops\307_crop_0.jpg
./dpjcrops\3114_crop_0.jpg
./dpjcrops\321_crop_0.jpg
./dpjcrops\3267_crop_0.jpg
./dpjcrops\3267_crop_1.jpg
./dpjcrops\3267_crop_2.jpg
./dpjcrops\3267_crop_3.jpg
./dpjcrops\3267_crop_4.jpg
./dpjcrops\3268_crop_0.jpg
./dpjcrops\3269_010_crop_0.jpg
./dpjcrops\3269_010_crop_1.jpg
./dpjcrops\3269_010_crop_2.jpg
./dpjcrops\3269_010_crop_3.jpg
./dpjcrops\3269_010_crop_4.jpg
./dpjcrops\3270_crop_0.jpg
./dpjcrops\3271_crop_0.jpg
./dpjcrops\3271_crop_1.jpg
./dpjcrops\3271_crop_2.jpg
./dpjcrops\3271_crop_3.jpg
./dpjcrops\3272_crop_0.jpg
./dpjcrops\3273_crop_0.jpg
./dpjcrops\3274_crop_0.jpg
./dpjcrops\3274_crop_1.jpg
./dpjcrops\3274_crop_2.jpg
./dpjcrops\3274_crop_3.jpg
./dpjcrops\3274_crop_4.jpg
./dpjcrops\3275_crop_0.jpg
./dpjcrops\3276_crop_0.jpg
./dpjcrops\3277_crop_0.jpg
./dpjcrops\3277_crop_1.jpg
./dpjcrops\3277_crop_2.jpg
./dpjcrops\3277_crop_3.jpg
./dpjcrops\3278_crop_0.jpg
./dpjcrops\3279_crop_0.jpg
./dpjcrops\3280_crop_0.jpg
./dpjcrops\3281_crop_0.jpg
./dpjcrops\3281_026_crop_0.jpg
./dpjcrops\3282_crop_0.jpg
./dpjcrops\3282_crop_1.jpg
./dpjcrops\3282_crop_2.jpg
./dpjcrops\3282_crop_3.jpg
./dpjcrops\3283_09_crop_0.jpg
./dpjcrops\3284_crop_0.jpg
./dpjcrops\3287_crop_0.jpg
./dpjcrops\3287_crop_1.jpg
./dpjcrops\3287_crop_2.jpg
./dpjcrops\3287_crop_3.jpg
./dpjcrops\3287_crop_4.jpg
./dpjcrops\3287_crop_5.jpg
./dpjcrops\3287_crop_6.jpg
./dpjcrops\3287_crop_7.jpg
./dpjcrops\3288_crop_0.jpg
./dpjcrops\3289_05_crop_0.jpg
./dpjcrops\3293_crop_0.jpg
./dpjcrops\3294_crop_0.jpg
./dpjcrops\3295_crop_0.jpg
./dpjcrops\3298_crop_0.jpg
./dpjcrops\3299_crop_0.jpg
./dpjcrops\3300_crop_0.jpg
./dpjcrops\3306_crop_0.jpg
./dpjcrops\3307_crop_0.jpg
./dpjcrops\3309_crop_0.jpg
./dpjcrops\3310_crop_0.jpg
./dpjcrops\3312_crop_0.jpg
./dpjcrops\3313_crop_0.jpg
./dpjcrops\3316_crop_0.jpg
./dpjcrops\3318_crop_0.jpg
./dpjcrops\3320_crop_0.jpg
./dpjcrops\3321_crop_0.jpg
./dpjcrops\3322_crop_0.jpg
./dpjcrops\3324_crop_0.jpg
./dpjcrops\3326_crop_0.jpg
./dpjcrops\3329_crop_0.jpg
./dpjcrops\3333_crop_0.jpg
./dpjcrops\3340_crop_0.jpg
./dpjcrops\335_crop_0.jpg
./dpjcrops\3355_crop_0.jpg
./dpjcrops\3388_crop_0.jpg
./dpjcrops\3389_crop_0.jpg
./dpjcrops\3390_crop_0.jpg
./dpjcrops\3405_crop_0.jpg
./dpjcrops\3410_crop_0.jpg
./dpjcrops\3419_crop_0.jpg
./dpjcrops\3432_crop_0.jpg
./dpjcrops\349_crop_0.jpg
./dpjcrops\3499_crop_0.jpg
./dpjcrops\3499_crop_1.jpg
./dpjcrops\3502_crop_0.jpg
./dpjcrops\3502_crop_1.jpg
./dpjcrops\3507_crop_0.jpg
./dpjcrops\3508_crop_0.jpg
./dpjcrops\3508_crop_1.jpg
./dpjcrops\3525_crop_0.jpg
./dpjcrops\357_crop_0.jpg
./dpjcrops\3573_crop_0.jpg
./dpjcrops\359_crop_0.jpg
./dpjcrops\3591_crop_0.jpg
./dpjcrops\3600_crop_0.jpg
./dpjcrops\3620_crop_0.jpg
./dpjcrops\3742_crop_0.jpg
./dpjcrops\3760_crop_0.jpg
./dpjcrops\3781_crop_0.jpg
./dpjcrops\3784_crop_0.jpg
./dpjcrops\3817_crop_0.jpg
./dpjcrops\3820_crop_0.jpg
./dpjcrops\3846_crop_0.jpg
./dpjcrops\3886_crop_0.jpg
./dpjcrops\3894_crop_0.jpg
./dpjcrops\393_crop_0.jpg
./dpjcrops\399_crop_0.jpg
./dpjcrops\406_crop_0.jpg
./dpjcrops\426_crop_0.jpg
./dpjcrops\426_crop_1.jpg
./dpjcrops\429_crop_0.jpg
./dpjcrops\488_crop_0.jpg
./dpjcrops\510_crop_0.jpg
./dpjcrops\574_crop_0.jpg
./dpjcrops\576_crop_0.jpg
./dpjcrops\581_crop_0.jpg
./dpjcrops\59_crop_0.jpg
./dpjcrops\594_crop_0.jpg
./dpjcrops\601_crop_0.jpg
./dpjcrops\617_crop_0.jpg
./dpjcrops\68_crop_0.jpg
./dpjcrops\69_crop_0.jpg
./dpjcrops\847_crop_0.jpg
./dpjcrops\992_crop_0.jpg
./dpjcrops\997_crop_0.jpg
56 changes: 56 additions & 0 deletions SmallObjectAugmentation/dpj_train.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
./dpjsamplecrop\125.jpg
./dpjsamplecrop\193.jpg
./dpjsamplecrop\273.jpg
./dpjsamplecrop\173.jpg
./dpjsamplecrop\194.jpg
./dpjsamplecrop\452.jpg
./dpjsamplecrop\465.jpg
./dpjsamplecrop\505.jpg
./dpjsamplecrop\876.jpg
./dpjsamplecrop\1527.jpg
./dpjsamplecrop\1076.jpg
./dpjsamplecrop\883.jpg
./dpjsamplecrop\1120.jpg
./dpjsamplecrop\1501.jpg
./dpjsamplecrop\717.jpg
./dpjsamplecrop\1118.jpg
./dpjsamplecrop\2124.jpg
./dpjsamplecrop\2233.jpg
./dpjsamplecrop\1869.jpg
./dpjsamplecrop\1976.jpg
./dpjsamplecrop\2019.jpg
./dpjsamplecrop\2213.jpg
./dpjsamplecrop\2067.jpg
./dpjsamplecrop\2032.jpg
./dpjsamplecrop\3046.jpg
./dpjsamplecrop\1903.jpg
./dpjsamplecrop\619.jpg
./dpjsamplecrop\3118.jpg
./dpjsamplecrop\3313_06.jpg
./dpjsamplecrop\16.jpg
./dpjsamplecrop\2189.jpg
./dpjsamplecrop\3296_03.jpg
./dpjsamplecrop\3277.jpg
./dpjsamplecrop\3268.jpg
./dpjsamplecrop\3274.jpg
./dpjsamplecrop\3275.jpg
./dpjsamplecrop\3270.jpg
./dpjsamplecrop\3278.jpg
./dpjsamplecrop\3271.jpg
./dpjsamplecrop\3281.jpg
./dpjsamplecrop\3291.jpg
./dpjsamplecrop\3301.jpg
./dpjsamplecrop\3296.jpg
./dpjsamplecrop\3325.jpg
./dpjsamplecrop\3304.jpg
./dpjsamplecrop\3323.jpg
./dpjsamplecrop\3284.jpg
./dpjsamplecrop\3309.jpg
./dpjsamplecrop\3677.jpg
./dpjsamplecrop\3685.jpg
./dpjsamplecrop\3858.jpg
./dpjsamplecrop\3712.jpg
./dpjsamplecrop\3880.jpg
./dpjsamplecrop\3800.jpg
./dpjsamplecrop\3864.jpg
./dpjsamplecrop\3787.jpg
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3266.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.5954545454545455 0.23928571428571427 0.05454545454545454 0.07857142857142857
5 changes: 5 additions & 0 deletions SmallObjectAugmentation/dpjimagetxt/3267.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
0 0.56171875 0.153125 0.035937500000000004 0.04375
0 0.95234375 0.12395833333333334 0.026562500000000003 0.04375
0 0.6570312500000001 0.140625 0.035937500000000004 0.03958333333333333
0 0.7296875 0.15208333333333332 0.034375 0.041666666666666664
0 0.846875 0.12395833333333334 0.034375 0.04791666666666666
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3268.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.5909090909090908 0.2357142857142857 0.06363636363636363 0.09999999999999999
5 changes: 5 additions & 0 deletions SmallObjectAugmentation/dpjimagetxt/3269.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
0 0.69453125 0.13958333333333334 0.035937500000000004 0.05
0 0.88203125 0.11979166666666667 0.0390625 0.05625
0 0.60078125 0.15416666666666667 0.0390625 0.058333333333333334
0 0.7671875 0.15208333333333332 0.03125 0.05
0 0.9851562500000001 0.121875 0.026562500000000003 0.052083333333333336
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3270.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.296 0.354 0.056 0.052000000000000005
4 changes: 4 additions & 0 deletions SmallObjectAugmentation/dpjimagetxt/3271.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
0 0.63828125 0.15416666666666667 0.035937500000000004 0.05
0 0.8 0.15520833333333334 0.034375 0.03958333333333333
0 0.73203125 0.14166666666666666 0.0328125 0.04583333333333333
0 0.9148437500000001 0.12395833333333334 0.0421875 0.04375
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3272.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.302 0.372 0.044 0.048
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3273.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.5931818181818181 0.23928571428571427 0.05909090909090909 0.09285714285714286
5 changes: 5 additions & 0 deletions SmallObjectAugmentation/dpjimagetxt/3274.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
0 0.56171875 0.15416666666666667 0.0328125 0.04583333333333333
0 0.8453125 0.12083333333333333 0.037500000000000006 0.04583333333333333
0 0.6585937500000001 0.140625 0.035937500000000004 0.04375
0 0.7273437500000001 0.15520833333333334 0.0328125 0.04791666666666666
0 0.9554687500000001 0.12395833333333334 0.0421875 0.04375
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3275.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.3 0.372 0.04 0.048
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3276.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.5909090909090908 0.2357142857142857 0.05454545454545454 0.08571428571428572
4 changes: 4 additions & 0 deletions SmallObjectAugmentation/dpjimagetxt/3277.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
0 0.9046875000000001 0.12083333333333333 0.040625 0.041666666666666664
0 0.7914062500000001 0.15416666666666667 0.0328125 0.04583333333333333
0 0.6273437500000001 0.153125 0.035937500000000004 0.03958333333333333
0 0.7210937500000001 0.14270833333333333 0.0328125 0.04791666666666666
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3278.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.296 0.352 0.04 0.048
1 change: 1 addition & 0 deletions SmallObjectAugmentation/dpjimagetxt/3279.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
0 0.302 0.374 0.044 0.044
Loading

0 comments on commit da93ae0

Please sign in to comment.