Skip to content

GitLab

F
food-pytorch-baseline
Commit 13ff79ef authored by spmohanty's avatar spmohanty
Browse files
Options

Update helper scripts

parent ad10db9a
Hide whitespace changes
Inline Side-by-side
Showing with 160 additions and 162 deletions
all_categories.json 0 → 100644
View file @ 13ff79ef
[{"id": 1565,
"name": "bread-wholemeal",
"name_readable": "Bread, wholemeal",
"supercategory": "food"},
{"id": 1010,
"name": "potatoes-steamed",
"name_readable": "Potatoes steamed",
"supercategory": "food"},
{"id": 1085,
"name": "broccoli",
"name_readable": "Broccoli",
"supercategory": "food"},
{"id": 2053,
"name": "butter",
"name_readable": "Butter",
"supercategory": "food"},
{"id": 1310,
"name": "hard-cheese",
"name_readable": "Hard cheese",
"supercategory": "food"},
{"id": 2578,
"name": "water",
"name_readable": "Water",
"supercategory": "food"},
{"id": 1154,
"name": "banana",
"name_readable": "Banana",
"supercategory": "food"},
{"id": 2620,
"name": "wine-white",
"name_readable": "Wine, white",
"supercategory": "food"},
{"id": 1566,
"name": "bread-white",
"name_readable": "Bread, white",
"supercategory": "food"},
{"id": 1151,
"name": "apple",
"name_readable": "Apple",
"supercategory": "food"},
{"id": 2939,
"name": "pizza-margherita-baked",
"name_readable": "Pizza, Margherita, baked",
"supercategory": "food"},
{"id": 1040,
"name": "salad-leaf-salad-green",
"name_readable": "Salad, leaf / salad, green",
"supercategory": "food"},
{"id": 1070,
"name": "zucchini",
"name_readable": "Zucchini",
"supercategory": "food"},
{"id": 2580,
"name": "water-mineral",
"name_readable": "Water, mineral",
"supercategory": "food"},
{"id": 2512,
"name": "coffee-with-caffeine",
"name_readable": "Coffee, with caffeine",
"supercategory": "food"},
{"id": 1056,
"name": "avocado",
"name_readable": "Avocado",
"supercategory": "food"},
{"id": 1069,
"name": "tomato",
"name_readable": "Tomato",
"supercategory": "food"},
{"id": 2131,
"name": "dark-chocolate",
"name_readable": "Dark chocolate",
"supercategory": "food"},
{"id": 2521,
"name": "white-coffee-with-caffeine",
"name_readable": "White coffee, with caffeine",
"supercategory": "food"},
{"id": 2022, "name": "egg", "name_readable": "Egg", "supercategory": "food"},
{"id": 1026,
"name": "mixed-salad-chopped-without-sauce",
"name_readable": "Mixed salad (chopped without sauce)",
"supercategory": "food"},
{"id": 1068,
"name": "sweet-pepper",
"name_readable": "Sweet pepper",
"supercategory": "food"},
{"id": 1022,
"name": "mixed-vegetables",
"name_readable": "Mixed vegetables",
"supercategory": "food"},
{"id": 2750,
"name": "mayonnaise",
"name_readable": "Mayonnaise",
"supercategory": "food"},
{"id": 1468,
"name": "rice",
"name_readable": "Rice",
"supercategory": "food"},
{"id": 1013,
"name": "chips-french-fries",
"name_readable": "Chips, french fries",
"supercategory": "food"},
{"id": 1078,
"name": "carrot",
"name_readable": "Carrot",
"supercategory": "food"},
{"id": 2738,
"name": "tomato-sauce",
"name_readable": "Tomato sauce",
"supercategory": "food"},
{"id": 1061,
"name": "cucumber",
"name_readable": "Cucumber",
"supercategory": "food"},
{"id": 2618,
"name": "wine-red",
"name_readable": "Wine, red",
"supercategory": "food"},
{"id": 1311,
"name": "cheese",
"name_readable": "Cheese",
"supercategory": "food"},
{"id": 1163,
"name": "strawberries",
"name_readable": "Strawberries",
"supercategory": "food"},
{"id": 2504,
"name": "espresso-with-caffeine",
"name_readable": "Espresso, with caffeine",
"supercategory": "food"},
{"id": 2498, "name": "tea", "name_readable": "Tea", "supercategory": "food"},
{"id": 1788,
"name": "chicken",
"name_readable": "Chicken",
"supercategory": "food"},
{"id": 2099, "name": "jam", "name_readable": "Jam", "supercategory": "food"},
{"id": 1032,
"name": "leaf-spinach",
"name_readable": "Leaf spinach",
"supercategory": "food"},
{"id": 1505,
"name": "pasta-spaghetti",
"name_readable": "Pasta, spaghetti",
"supercategory": "food"},
{"id": 1058,
"name": "french-beans",
"name_readable": "French beans",
"supercategory": "food"},
{"id": 1554,
"name": "bread-whole-wheat",
"name_readable": "Bread, whole wheat",
"supercategory": "food"}]
build.sh 100644 → 100755
View file @ 13ff79ef
#!/bin/bash
docker build -t $IMAGE_NAME
docker build -t $IMAGE_NAME .
debug.sh 100644 → 100755
View file @ 13ff79ef
File mode changed from 100644 to 100755
infer.py
View file @ 13ff79ef
......@@ -15,15 +15,12 @@ from mmdet.models import build_detector
import glob
import json
test_images_path = os.getenv("AICROWD_TEST_IMAGES_PATH", False)
TEST_IMAGES_PATH = os.getenv("AICROWD_TEST_IMAGES_PATH", False)
# Build a sample test_annotations file
annotations = {'categories': [], 'info': {}, 'images': []}
for item in glob.glob(test_images_path+'/*.jpg'):
for item in glob.glob(TEST_IMAGES_PATH+'/*.jpg'):
image_dict = dict()
img = mmcv.imread(item)
height,width,__ = img.shape
id = int(os.path.basename(item).split('.')[0])
......@@ -32,158 +29,8 @@ for item in glob.glob(test_images_path+'/*.jpg'):
image_dict['width'] = width
image_dict['height'] = height
annotations['images'].append(image_dict)
annotations['categories'] = [{'id': 1565,
'name': 'bread-wholemeal',
'name_readable': 'Bread, wholemeal',
'supercategory': 'food'},
{'id': 1010,
'name': 'potatoes-steamed',
'name_readable': 'Potatoes steamed',
'supercategory': 'food'},
{'id': 1085,
'name': 'broccoli',
'name_readable': 'Broccoli',
'supercategory': 'food'},
{'id': 2053,
'name': 'butter',
'name_readable': 'Butter',
'supercategory': 'food'},
{'id': 1310,
'name': 'hard-cheese',
'name_readable': 'Hard cheese',
'supercategory': 'food'},
{'id': 2578,
'name': 'water',
'name_readable': 'Water',
'supercategory': 'food'},
{'id': 1154,
'name': 'banana',
'name_readable': 'Banana',
'supercategory': 'food'},
{'id': 2620,
'name': 'wine-white',
'name_readable': 'Wine, white',
'supercategory': 'food'},
{'id': 1566,
'name': 'bread-white',
'name_readable': 'Bread, white',
'supercategory': 'food'},
{'id': 1151,
'name': 'apple',
'name_readable': 'Apple',
'supercategory': 'food'},
{'id': 2939,
'name': 'pizza-margherita-baked',
'name_readable': 'Pizza, Margherita, baked',
'supercategory': 'food'},
{'id': 1040,
'name': 'salad-leaf-salad-green',
'name_readable': 'Salad, leaf / salad, green',
'supercategory': 'food'},
{'id': 1070,
'name': 'zucchini',
'name_readable': 'Zucchini',
'supercategory': 'food'},
{'id': 2580,
'name': 'water-mineral',
'name_readable': 'Water, mineral',
'supercategory': 'food'},
{'id': 2512,
'name': 'coffee-with-caffeine',
'name_readable': 'Coffee, with caffeine',
'supercategory': 'food'},
{'id': 1056,
'name': 'avocado',
'name_readable': 'Avocado',
'supercategory': 'food'},
{'id': 1069,
'name': 'tomato',
'name_readable': 'Tomato',
'supercategory': 'food'},
{'id': 2131,
'name': 'dark-chocolate',
'name_readable': 'Dark chocolate',
'supercategory': 'food'},
{'id': 2521,
'name': 'white-coffee-with-caffeine',
'name_readable': 'White coffee, with caffeine',
'supercategory': 'food'},
{'id': 2022, 'name': 'egg', 'name_readable': 'Egg', 'supercategory': 'food'},
{'id': 1026,
'name': 'mixed-salad-chopped-without-sauce',
'name_readable': 'Mixed salad (chopped without sauce)',
'supercategory': 'food'},
{'id': 1068,
'name': 'sweet-pepper',
'name_readable': 'Sweet pepper',
'supercategory': 'food'},
{'id': 1022,
'name': 'mixed-vegetables',
'name_readable': 'Mixed vegetables',
'supercategory': 'food'},
{'id': 2750,
'name': 'mayonnaise',
'name_readable': 'Mayonnaise',
'supercategory': 'food'},
{'id': 1468,
'name': 'rice',
'name_readable': 'Rice',
'supercategory': 'food'},
{'id': 1013,
'name': 'chips-french-fries',
'name_readable': 'Chips, french fries',
'supercategory': 'food'},
{'id': 1078,
'name': 'carrot',
'name_readable': 'Carrot',
'supercategory': 'food'},
{'id': 2738,
'name': 'tomato-sauce',
'name_readable': 'Tomato sauce',
'supercategory': 'food'},
{'id': 1061,
'name': 'cucumber',
'name_readable': 'Cucumber',
'supercategory': 'food'},
{'id': 2618,
'name': 'wine-red',
'name_readable': 'Wine, red',
'supercategory': 'food'},
{'id': 1311,
'name': 'cheese',
'name_readable': 'Cheese',
'supercategory': 'food'},
{'id': 1163,
'name': 'strawberries',
'name_readable': 'Strawberries',
'supercategory': 'food'},
{'id': 2504,
'name': 'espresso-with-caffeine',
'name_readable': 'Espresso, with caffeine',
'supercategory': 'food'},
{'id': 2498, 'name': 'tea', 'name_readable': 'Tea', 'supercategory': 'food'},
{'id': 1788,
'name': 'chicken',
'name_readable': 'Chicken',
'supercategory': 'food'},
{'id': 2099, 'name': 'jam', 'name_readable': 'Jam', 'supercategory': 'food'},
{'id': 1032,
'name': 'leaf-spinach',
'name_readable': 'Leaf spinach',
'supercategory': 'food'},
{'id': 1505,
'name': 'pasta-spaghetti',
'name_readable': 'Pasta, spaghetti',
'supercategory': 'food'},
{'id': 1058,
'name': 'french-beans',
'name_readable': 'French beans',
'supercategory': 'food'},
{'id': 1554,
'name': 'bread-whole-wheat',
'name_readable': 'Bread, whole wheat',
'supercategory': 'food'}]
json.dump(annotations, open('test.json', 'w'))
annotations['categories'] = json.loads(open("all_categories.json").read())
json.dump(annotations, open('test_annotations.json', 'w'))
def single_gpu_test(model, data_loader, show=False):
......@@ -318,8 +165,8 @@ def main():
torch.backends.cudnn.benchmark = True
cfg.model.pretrained = None
cfg.data.test.test_mode = True
cfg.data.test.ann_file = 'test.json'
cfg.data.test.img_prefix = test_images_path
cfg.data.test.ann_file = 'test_annotations.json'
cfg.data.test.img_prefix = TEST_IMAGES_PATH
# init distributed env first, since logger depends on the dist info.
if args.launcher == 'none':
......
run.sh
View file @ 13ff79ef
#!/bin/bash
python tools/test.py configs/htc_r50.py models/epoch_22.pth --json_out $AICROWD_PREDICTIONS_OUTPUT_PATH
python infer.py configs/htc_r50.py models/epoch_22.pth --json_out $AICROWD_PREDICTIONS_OUTPUT_PATH
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment